一个节点的子孙节点,指这个节点的子节点,以及子节点的子节点,等等。 例如,叶子节点的子孙节点个数为0,根节点的子孙节点个数为n-1。 给你一棵n个节点的有根树,节点标号为1~n,1号节点为根节点。 现在询问节点k有多少个子孙节点。 输入 第一行两个整数n,k。 以后n行,每行若干个整数。第i行,第一个数为mi,表示节点i的子节点个数。紧接着mi个整数,表示节点i子节点的编号。保证父节点编号小于子节点。 输出 一行,一个整数,表示节点k的子孙节点个数。c++代码
时间: 2024-04-15 08:28:46 浏览: 141
//计算二叉树叶子结点的个数-C语言数据结构——树
```cpp
#include <iostream>
#include <vector>
using namespace std;
vector<int> descendants;
vector<vector<int>> tree;
int countDescendants(int node) {
int count = 0;
for (int child : tree[node]) {
count += countDescendants(child) + 1;
}
return count;
}
int main() {
int n, k;
cin >> n >> k;
tree.resize(n + 1);
descendants.resize(n + 1);
for (int i = 1; i <= n; i++) {
int m;
cin >> m;
for (int j = 0; j < m; j++) {
int child;
cin >> child;
tree[i].push_back(child);
}
}
int numDescendants = countDescendants(k);
cout << numDescendants << endl;
return 0;
}
```
这段代码通过递归地计算节点的子孙节点个数。首先,我们定义了一个 `descendants` 数组来保存每个节点的子孙节点个数。然后,我们使用一个二维向量 `tree` 来表示树的结构,其中 `tree[i]` 表示节点 i 的子节点列表。在主函数中,我们首先读取输入数据,构建树的结构。然后,我们调用 `countDescendants` 函数来计算节点 k 的子孙节点个数。在 `countDescendants` 函数中,我们通过递归地遍历节点的子节点,并累加子孙节点的个数。最后,我们输出节点 k 的子孙节点个数。
阅读全文