『P1122』最大子树和

题目传送门

  输入样例

7
-1 -1 -1 1 1 1 0
1 4
2 5
3 6
4 7
5 7
6 7

   输出样例

3

  code

#include<bits/stdc++.h>
using namespace std;
int fro[100005],cnt,n,v[100005],ans=0;
struct node{int to,nxt;}a[100000];
void add(int x,int y) {
    a[++cnt].to=y; a[cnt].nxt=fro[x]; fro[x]=cnt;
}

int dfs(int u,int fa) {
    int f=v[u];
    for(int i=fro[u];i!=0;i=a[i].nxt) {
        int v=a[i].to;
        if(v!=fa) f+=max(dfs(v,u),0);
    }
    ans=max(ans,f);
    return f;
}

int main() {
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&v[i]);
    for(int i=1;i<n;i++) {
        int x,y; scanf("%d%d",&x,&y);
        add(x,y); add(y,x);
    }
    dfs(1,0);
    printf("%d",ans);
}
/*
树形DP
子树的取舍：取（加上价值），舍（价值为0）
*/