Codeforces Round #379 (Div. 2) E. Anton and Tree

链接：http://codeforces.com/contest/734/problem/E

题意：给定一棵树，每个节点为黑色或白色，有一种操作：现在一个节点然后将与其联通的同颜色的节点一起变一种颜色（黑变白，白变黑），求最少多少次操作可以使得整棵树变成一个颜色。

分析：因为可以一个同颜色的联通块一起变，我们将所有同颜色的联通块缩点，然后我们会发现树变成了全是黑白交替的。那么最小操作次数就是从中心向外变颜色直到全部一个颜色即可，中心就是直径中间咯，答案就是直接的长度/2。

代码：

#include<map>
#include<set>
#include<stack>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
typedef double db;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
const db eps=1e-5;
const int N=2e5+10;
const int M=4e5+10;
const ll MOD=1000000007;
const int mod=1000000007;
const int MAX=1000000010;
const double pi=acos(-1.0);
int md,ans,c[N],deep[N];
int tot,u[N],v[M],pre[M];
void add(int x,int y) {
    v[tot]=y;pre[tot]=u[x];u[x]=tot++;
    v[tot]=x;pre[tot]=u[y];u[y]=tot++;
}
int a[N],b[N],f[N],k[N];
int find_f(int x) {
    return f[x]==x ? x:f[x]=find_f(f[x]);
}
void unio(int x,int y) {
    int fx=find_f(x),fy=find_f(y);
    if (fx!=fy) f[fx]=fy;
}
void dfs(int x,int y) {
    deep[x]=deep[y]+1;
    if (deep[x]>ans) ans=deep[x],md=x;
    for (int i=u[x];~i;i=pre[i])
    if (v[i]!=y) dfs(v[i],x);
}
int main()
{
    int i,g,n;
    scanf("%d", &n);
    for (i=1;i<=n;i++) scanf("%d", &c[i]);
    for (i=1;i<=n;i++) f[i]=i;
    for (i=1;i<n;i++) {
        scanf("%d%d", &a[i], &b[i]);
        if (c[a[i]]==c[b[i]]) unio(a[i],b[i]);
    }
    for (g=0,i=1;i<=n;i++) {
        f[i]=find_f(i);
        if (f[i]==i) k[i]=++g;
    }
    for (i=1;i<=n;i++) k[i]=k[f[i]];

    tot=0;memset(u,-1,sizeof(u));
    for (i=1;i<n;i++)
    if (k[a[i]]!=k[b[i]]) add(k[a[i]],k[b[i]]);

    md=ans=deep[0]=0;
    dfs(1,0);
    ans=0;dfs(md,0);

    printf("%d\n", ans/2);
    return 0;
}