P1122最大子树和 （树形动规）

入门题目

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
const int MAXN=160010;
struct Node
{
	int next,to;
}edge[MAXN<<2];
int n,f[MAXN],vis[MAXN],head[MAXN],cnt,flower[MAXN];
using namespace std;
inline void Add_Edge(int u,int v)
{
	edge[++cnt]=(Node) {head[u],v};
	head[u]=cnt;
}
void dfs(int u)
{
	f[u]=flower[u];
	vis[u]=1;
	for(int i=head[u];i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(vis[v]==0)
		{
			dfs(v);
			if(f[v]>0) f[u]+=f[v];
		}
	}
} 
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&flower[i]);
	for(int i=1;i<=n-1;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		Add_Edge(x,y);
		Add_Edge(y,x);
	}
	dfs(1);
	int ans=f[1];
	for(int i=1;i<=n;i++) ans=max(ans,f[i]);
	printf("%d",ans);
	return 0;
 }