两个问题

#include<bits/stdc++.h>
using namespace std;
const int N=6005;
int n,a,b,c,dp[N],g[N];
int ha[N],root;
vector<int>q[N];
void dfs(int now,int father)
{
	dp[now]=ha[now];
	g[now]=0;
	for(int i=0;i<q[now].size();i++)
	{
		int son=q[now][i];
		if(son==father) continue;
		dfs(son,now);
		dp[now]+=g[son];
		g[now]+=max(dp[son],g[son]);
	}
}
int mian()
{
	cin>>n;
	root=n*(n+1)/2;
	for(int i=1;i<=n;i++)
	{
		cin>>ha[i];
	}
	for(int i=1;i<n;i++)
	{
		cin>>a>>b;
		q[a].push_back(b);
		q[b].push_back(a);
	}
	dfs(root,-1);
	cout<<max(dp[root],g[root]);
	return 0;
}