題目背景

墨家家主有棵樹。
題目描述

給定一個保存樹節點信息的數據結構，它包含了樹節點唯一的 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; }

創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的树的节点值之和的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。