【生命之树】

题目

思路

求联通区域中的最大和值

代码

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+10, M = N << 1;
const int null = -0x3f3f3f3f;
long long w[N];
int h[N], e[M], ne[M], idx;
void add(int a, int b)  // 添加一条边a->b
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}
int n;
long long ans = null;
long long dfs(int f, int u)
{
    long long bsx = 0;  //branches sum max
    
    for(int k = h[u]; ~k; k = ne[k])
    {
        int j = e[k];
        if(j == f) continue;
        long long d = dfs(u, j);
        if(d > 0) bsx += d;
    }
    ans = max(ans, bsx+w[u]);
    return bsx+w[u];
}
int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++)
    {
        cin >> w[i];
    }
    memset(h, -1, sizeof h);
    for(int i = 0; i < n-1; i ++)
    {
        int a, b;
        cin >> a >> b;
        add(a, b); add(b, a);
    }
    dfs(0, 1);
    cout << ans;
    return 0;
}

注意

题目说节点绝对值不大于，说明 bsx 和 函数返回值会爆 int，同时由于使用了 max 函数，所以w[ ]  和 ans 也需要开 long long