题目背景
墨家家主有棵树。
题目描述
给定一个保存树节点信息的数据结构,它包含了树节点唯一的 id ,树节点值 和 直系子节点的 id 。
比如,树节点1是树节点2的父节点,树节点2是树节点3的父节点。他们相应的树节点值为 9 , 4 , 1 。那么树节点1的数据结构是 [1, 9, [2]] ,树节点2的数据结构是 [2, 4, [3]] ,树节点3的数据结构是 [3, 1, []] 。注意虽然树节点3也是树节点1的一个子节点,但是由于并不是直系子节点,因此没有体现在树节点1的数据结构中。
现在输入一棵树的所有树节点信息,以及单个树节点id,返回这个树节点和它所有子节点的节点值之和。
输入格式
第一行输入一个整数N,代表共有N个树节点。
接下来输入N个树节点,每个树节点第一行输入它的id,第二行输入节点的值,第三行输入直系子节点的个数M,第四行输入M个直系子节点的id值。
最后一行输入一个树节点id。
输出格式
一个整数,最后一行输入的树节点的和它所有子节点的节点值之和。
输入输出样例
输入
3
1
5
2
2 3
2
3
0
3
3
0
1
输出
11
说明/提示
树节点 1 自身的节点值是 5 ,他有两个直系子节点 2 和 3 ,而且 2 和 3 的节点值均为 3 。因此树节点 1 的总节点值是 5 + 3 + 3 = 11 。
一个树节点最多有一个直系父节点,但是可以有多个直系子节点。
节点数量不超过 2000 。
解题思路:
正常模拟一棵树!!!但是用vector容器,在规模较大的时候,效率比数组慢很多,观察下面的数据点可以知道,最后两个数据点,用vector容器的写法是过不了的。
代码如下:
#include <iostream>
#include <vector>
using namespace std;
const int N = 1010;
vector<int>tnode[N];
int w[N];
int ans;
void dfs(int node)
{ans+=w[node];for (int i = 0;i<tnode[node].size();i++){dfs(tnode[node][i]);}
}int main()
{int cnt;cin>>cnt;while(cnt--){int id,chiln;cin>>id;cin>>w[id];cin>>chiln;if (chiln==0) continue;else{for (int i = 1;i<=chiln;i++){int chilid;cin>>chilid;tnode[id].push_back(chilid);}}}int root;cin>>root;dfs(root);cout<<ans<<endl;return 0;
}
代码如下:
#include <iostream>
using namespace std;
const int N = 2010;
int ans;struct node {int w;int ch[N];int size;
};node tnode [N];void dfs(int root) {ans += tnode[root].w;for (int i = 1; i <= tnode[root].size; i++) {dfs(tnode[root].ch[i]);}
}int main() {int cnt;cin >> cnt;while (cnt--) {int id, n;cin >> id;cin >> tnode[id].w;cin >> n;if (n == 0)continue;else {for (int i = 1; i <= n; i++) {cin >> tnode[id].ch[i];}tnode[id].size = n;}}int root;cin >> root;dfs(root);cout << ans << endl;return 0;
}