⑨0pts WA on #5 悬1关

#include<bits/stdc++.h>
using namespace std;
int c[100005],dp[100005],ans=1;
vector<int> ve[100005];
void dfs(int x,int f){
	vector<int> vt;
	dp[x]=1;
	for(int i=0;i<ve[x].size();i++){
		int d=ve[x][i];
		if(d==f)continue;
		//cout<<d<<' '<<c[d]<<' '<<x<<' '<<c[x]<<endl;
		dfs(d,x);
		if(c[d]!=c[x]){
			dp[x]=max(dp[x],dp[d]+1);
			vt.push_back(dp[d]);
		}
		//cout<<dp[x]<<endl;
	}
	if(vt.empty()) return;
    sort(vt.begin(),vt.end());
	if(vt.size()==1) ans=max(ans,vt[0]+1);
	else ans=max(ans,vt[0]+vt[1]+1);
}
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>c[i];
	}
	for(int i=1;i<=n-1;i++){
		int u,v;
		cin>>u>>v;
		ve[u].push_back(v);
		ve[v].push_back(u);
	}
	dfs(1,0);
	cout<<ans;
return 0;}