思路题，树的直径（Anton and Tree，cf 734E）

给你一棵树，节点的颜色非黑即白，因此树被分成很多联通块，每个联通块的颜色非黑即白，每次操作可以改变一个联通块的颜色，问你至少要操作几次才能使得所有节点颜色相同。方法就是先对联通块缩点建树，然后求出新树的直径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;
}