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