60pts求调

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=6e3+10;
vector<int> g[N];
int n,a[N];
int dp[N][2],fa[N];
int u,v;
int k;
void dfs(int x)
{
	dp[x][0]=0,dp[x][1]=a[x];
	for(auto v:g[x])
	{
		dfs(v);
		dp[x][0]+=max(dp[v][0],dp[v][1]);
		dp[x][1]+=dp[v][0];
	}
}
signed main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<n;i++)
    {
    	cin>>u>>v;
    	g[u].push_back(v);
	}
	for(int i=1;i<=n;i++)
	{
		if(!fa[i])
		{
			k=i;
			break;
		}
	}
	dfs(k);
	cout<<max(dp[k][0],dp[k][1]);
    return 0;
}