Luogu P1122 最大子树和

Luogu P1122 最大子树和

维护DP数组$f$，其中$f[x]$表示以$x$为根节点的子数的美观程度。
则对于任意节点$u$，有$f[u]=\sum f[v](v\in u的子树)$。
最后每次退出DFS的时候更新一下$ans$即可。

#include<bits/stdc++.h>
#define N 16010

using namespace std;

int n,cnt,ans;
int a[N],head[N],f[N];

struct node {
	int nxt,to;
}edge[N*2];

void addEdge(int u,int v) {
	edge[++cnt]=(node){head[u],v};
	head[u]=cnt;
	return;
}

void Read() {
	scanf("%d",&n);
	for(int i=1;i<=n;i++) {
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=n-1;i++) {
		int a,b;
		scanf("%d%d",&a,&b);
		addEdge(a,b);
		addEdge(b,a);
	}
	return;
}

void DFS(int x,int fa) {
	f[x]=a[x];
	for(int i=head[x];i;i=edge[i].nxt) {
		int v=edge[i].to;
		if(v!=fa) {
			DFS(v,x);
			f[x]+=max(0,f[v]);
		}
	}
	ans=max(ans,f[x]);
	return;
}

void Print() {
	printf("%d",ans);
}

int main()
{
	Read();
	DFS(1,0);
	Print();
	return 0;
}