全RE,求大佬帮条,玄关
查看原帖
全RE,求大佬帮条,玄关
1294257
zyx_Henry楼主2024/11/22 16:40
#include<bits/stdc++.h>
using namespace std;
int nxt[200005],to[200005],head[100005],d[100005][2],tot,col[100005],ans;
void add_edge(int u,int v)
{
	nxt[++tot]=head[u];
	to[tot]=v;
	head[u]=tot;
}
int dfs(int u,int pre)
{
	int i,v;
	for(i=head[u];i;i=nxt[i])
	{
		v=to[i];
		if(v==pre)
			continue;
		dfs(v,u);
		if(col[u]^col[v])
		{
			if(d[v][0]+1>d[u][0])
			{
				d[u][1]=d[u][0];
				d[u][0]=d[v][0]+1;
			}
			else if(d[v][0]+1>d[u][1])
				d[u][1]=d[v][0]+1;
		}
	}
	ans=max(ans,d[u][0]+d[u][1]);
}
int main()
{
	memset(d,0,sizeof(d));
	int n,i,a,b;
	scanf("%d",&n);
	for(i=1;i<=n;++i)
		scanf("%d",&col[i]);
	for(i=0;i<n-1;++i)
	{
		scanf("%d %d",&a,&b);
		add_edge(a,b);
		add_edge(b,a);
	}
	dfs(1,0);
	printf("%d",ans+1);
	return 0;
}
2024/11/22 16:40
加载中...