[蓝桥杯2015初赛]生命之树 （dfs）

代码：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<string>
using namespace std;
#define ll long long
const int  maxn=1e5+5;
#include<vector>
int q[maxn];  ///每个点的权重 
ll qq[maxn];   ///每个点作为根节点时能得到的最大权和 
vector<int>v[maxn];  ///邻接表 
ll ans;
void dfs(int root,int fa)  ///以root为根，算出最大权和 
{
	qq[root]=(ll)q[root];
	for(int i=0;i<v[root].size();i++)
	{
		int son=v[root][i];  ///root的一个孩子 
		if(son!=fa)
		{
			dfs(son,root);
			if(qq[son]>0)
			{
			    qq[root]+=qq[son]; 
			}
		}
	}
	if(qq[root]>ans)
	{
		ans=qq[root];
	}
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&q[i]);
	}
	for(int i=1;i<=n-1;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		v[x].push_back(y);
		v[y].push_back(x);
	}
	ans=0;
	dfs(1,0);
	printf("%lld\n",ans);
	return 0;
}