码迷,mamicode.com
首页 > 编程语言 > 详细

C++求树子节点权重最大的和

时间:2020-07-08 20:01:38      阅读:78      评论:0      收藏:0      [点我收藏+]

标签:height   mic   tor   for   strong   c++   ima   子节点   lock   

 

#include <iostream>
#include <vector>

using namespace std;
int n;
const int MaxN = 1e5;
long long w[MaxN + 1];
long long ans;
vector<int> g[MaxN + 1];

void dfs(int root, int fa) {
    for (int i = 0; i < g[root].size(); ++i) {
        int son = g[root][i];
        if (son != fa) {
            dfs(son, root);
            if (w[son] > 0)
                w[root] += w[son];
        }
    }
    if (w[root] > ans) ans = w[root];
}

void _cin() {
    std::ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n;
    for (int i = 1; i <= n; ++i) {
        cin >> w[i];
    }
    for (int j = 0; j < n - 1; ++j) {
        int v, u;
        cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
}

int main() {
    _cin();
    dfs(1, 0);
    cout << ans << endl;
    return 0;
}

技术图片

C++求树子节点权重最大的和

标签:height   mic   tor   for   strong   c++   ima   子节点   lock   

原文地址:https://www.cnblogs.com/HDawn/p/13268527.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!