0 分求条

#include<iostream>
using namespace std;
const int N=6e3+5;
int n,a[N],f[N][2];
bool mp[N][N];
void dfs(int x,int fa){
	f[x][1]=f[x][0]=a[x];
	for(int i=1;i<=n;i++){
		if(mp[x][i]&&i-fa){
			dfs(i,x);
			f[x][0]+=f[i][1];
			f[x][1]+=max(f[i][0],f[i][1]);
		}
	}
	f[x][0]-=a[x];
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	for(int i=1,x,y;i<n;i++){
		cin>>x>>y;
		mp[x][y]=1;
	}
	for(int i=1;i<=n;i++){
		for(int j=i+1;j<=n;j++){
			mp[j][i]|=mp[i][j];
			mp[i][j]|=mp[j][i];
			mp[j][i]|=mp[i][j];
		}
	}
	dfs(1,1);
	cout<<max(f[1][1],f[1][0])<<'\n';
	return 0;
}