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

2021-04-23 21:27

阅读:612

 

#include
#include

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

void dfs(int root, int fa) {
    for (int i = 0; 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         cin >> w[i];
    }
    for (int j = 0; j         int v, u;
        cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
}

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

上一篇:Java正则总结

下一篇:Python学习之爬虫


评论


亲,登录后才可以留言!