交暴力上去都会 WA。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int dep[maxn],fa[maxn][17],col[maxn];
struct edge{
	int y,next;
}e[maxn<<1];
int cnt,elast[maxn];
void add(int x,int y){
	e[++cnt].y=y;
	e[cnt].next=elast[x];
	elast[x]=cnt;
}
void dfs(int u,int f){
	fa[u][0]=f;
	dep[u]=dep[f]+1;
	for(int i=elast[u];i;i=e[i].next){
		int v=e[i].y;
		if(v!=f)dfs(v,u);
	}
}
int lca(int u,int v){
	if(dep[u]<dep[v])swap(u,v);
	for(int i=16;i>=0;i--){
		if(dep[fa[u][i]]>=dep[v]){
			u=fa[u][i];
		}
	}
	if(u==v)return u;
	for(int i=16;i>=0;i--){
		if(fa[u][i]!=fa[v][i]){
			u=fa[u][i];
			v=fa[v][i];
		}
	}
	return fa[u][0];
}
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d",&col[i]);
	for(int i=1;i<n;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	dfs(1,0);
	for(int j=1;j<=16;j++){
		for(int i=1;i<=n;i++){
			fa[i][j]=fa[fa[i][j-1]][j-1];
		}
	}
	int ans=0;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(col[i]!=col[j]){
				ans=max(ans,dep[i]+dep[j]-2*dep[lca(i,j)]);
			}
		}
	}
	printf("%d",ans);
	return 0;
}