洛谷 P1122 最大子树和 （树形dp）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/imzxww/article/details/81669978

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;
}