求助

#include <bits/stdc++.h>
using namespace std;
int n,x,y;
vector<int> e[100005];
long long f[100005][2],ans;
int color[100005];
void dfs(int x,int fath){
	f[x][color[x]]=0;
	long long maxw=-1e9,maxb=-1e9;
	if(color[x]==1) maxb=0;
	else maxw=0;
	for(int i=0;i<e[x].size();i++){
		if(fath==e[x][i]) continue;
		dfs(e[x][i],x);
		long long t;
		t=max(maxw,f[e[x][i]][0]+1);
		ans=max(ans,t+maxb);
		maxb=max(maxb,f[e[x][i]][1]+1);
		ans=max(ans,maxb+maxw);
		maxw=t;
	}
	f[x][0]=max(maxw,f[x][0]);
	f[x][1]=max(maxb,f[x][1]);
	return ;
}
int main(){
	memset(f,-0x3f,sizeof(f));
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&color[i]);
	for(int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		e[x].push_back(y);
		e[y].push_back(x);
	}
	dfs(1,-1);
	printf("%lld",ans);
}