CF734E Anton and Tree（树的直径）

我们每次相当于给一个极大同色连通块变色。
因此把同色连通块缩点。得到一棵树，以任意点为根，我们就得到了一颗邻层异色的树。我们只要从根开始，不断变色就可以把整棵树缩成一个点，需要最大深度次。
而我们希望最小，也就是要树的最大深度最小。
以树的直径的中点为根时，最大深度最小，为len+1/2。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,col[N],h[N],num=1,bel[N],tot=0,mx[N],ans=0;
struct edge{
    int to,next;
}data[N<<1];
void dfs(int x){
    bel[x]=tot;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(bel[y]||col[y]!=col[x]) continue;dfs(y);
    }
}
inline void add(int x,int y){
    data[++num].to=y;data[num].next=h[x];h[x]=num;
    data[++num].to=x;data[num].next=h[y];h[y]=num;
}
inline void dp(int x,int Fa){
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa) continue;dp(y,x);
        ans=max(ans,mx[x]+mx[y]+1);mx[x]=max(mx[x],mx[y]+1);
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();for(int i=1;i<=n;++i) col[i]=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }for(int i=1;i<=n;++i) if(!bel[i]) ++tot,dfs(i);
    memset(h,0,sizeof(h));int num1=num;num=0;
    for(int i=2;i<=num1;i+=2){
        int x=data[i].to,y=data[i^1].to;
        if(bel[x]!=bel[y]) add(bel[x],bel[y]);
    }dp(1,0);
    printf("%d\n",ans+1>>1);
    return 0;
}