https://www.luogu.org/problemnew/show/P1122
思路:
如果子树和为负的话肯定删去,那么删除标记,更新最大值。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e6+10;
vector<int>e[maxn];
int w[maxn];
int ma;
int sz[maxn];
int cnt[maxn];
void dfs(int x,int pre)
{
sz[x]=w[x];
for(int i=0;i<e[x].size();i++)
{
int v=e[x][i];
if(v==pre)continue;
dfs(v,x);
}
for(int i=0;i<e[x].size();i++)
{
int v=e[x][i];
if(v==pre||cnt[v])continue;
sz[x]+=sz[v];
}
if(sz[x]<0)cnt[x]=1;
ma=max(sz[x],ma);
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&w[i]);
}
int x,y;
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
e[x].push_back(y);
e[y].push_back(x);
}
ma=0;
dfs(1,-1);
printf("%d\n",ma);
return 0;
}