给你一棵树,节点的颜色非黑即白,因此树被分成很多联通块,每个联通块的颜色非黑即白,每次操作可以改变一个联通块的颜色,问你至少要操作几次才能使得所有节点颜色相同。方法就是先对联通块缩点建树,然后求出新树的直径M,答案就是(M+1)/2。
讲具体一点就是新树是黑白相间的,我们只需要找到树的中心u,然后对这个节点进行(M+1)/2次操作,纯色的联通块就会一圈一圈的扩大,在(M+1)/2操作后扩大到边界,此时所有节点颜色相同。
树的直径是M,直径上有M+1个点。
代码
#include<bits/stdc++.h>
#define maxn 200010
using namespace std;
//
//edges
struct Edge
{
int to,next;
};
struct star
{
Edge edges[maxn<<1];
int head[maxn],tot;
void init()
{
memset(head,-1,sizeof(head));
tot=0;
}
void add(int from,int to)
{
edges[tot].to=to;
edges[tot].next=head[from];
head[from]=tot++;
}
}M1,M2;
/
int n,c[maxn];
int id[maxn];
int vis[maxn];
int num;
void dfs(int u)
{
vis[u]=1;
id[u]=num;
for(int i=M1.head[u];~i;i=M1.edges[i].next)
{
int v=M1.edges[i].to;
if(!vis[v]&&c[u]==c[v])
dfs(v);
}
}
int dp[maxn];
int M;
void dfs2(int u,int f)
{
dp[u]=0;
for(int e=M2.head[u];~e;e=M2.edges[e].next)
{
int v=M2.edges[e].to;
if(v==f) continue;
dfs2(v,u);
M=max(M,dp[u]+dp[v]+1);
dp[u]=max(dp[u],dp[v]+1);
}
}
int main()
{
M1.init();
M2.init();
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&c[i]);
int u,v;
for(int i=1;i<n;i++)
{
scanf("%d %d",&u,&v);
M1.add(u,v);
M1.add(v,u);
}
for(int i=1;i<=n;i++)
if(!vis[i])
{
num++;
dfs(i);
}
for(int u=1;u<=n;u++)
for(int i=M1.head[u];~i;i=M1.edges[i].next)
{
int v=M1.edges[i].to;
if(id[u]!=id[v])
M2.add(id[u],id[v]);
}
M=0;
dfs2(1,0);
printf("%d\n",(M+1)/2);
return 0;
}