【生命之树】
题目
思路
求联通区域中的最大和值
代码
#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 maxfor(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