codeforces 734 E. Anton and Tree（连通分量缩点+树的直径）

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

思路：先缩点，变成黑白相间的一棵树，然后找出树的直径d，每改变一个节点的颜色，相当于将一整块变成一个新的连通分量，画张图可以看出来，答案就是(d+1)/2;

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define pb push_back
#define lc (d<<1)
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
int n,x,y,a[200008],deep,pos;
vector<int>g[200008],ne[200008];
bool vis[200008];
void dfs1(int x,int y)//连通分量缩点
{
    if(vis[x])  return;
    vis[x]=1;
    if(a[x]!=a[y])
    {
        ne[x].pb(y);
        ne[y].pb(x);
        y=x;
    }
    for(int i=0;i<g[x].size();i++)
    {
        dfs1(g[x][i],y);
    }
}
void dfs2(int d,int x)//两次DFS求树的直径
{
    if(vis[x])  return;
    vis[x]=1;
    if(d>deep)  deep=d,pos=x;
    for(int i=0;i<ne[x].size();i++)
    {
        dfs2(d+1,ne[x][i]);
    }
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    cin>>n;
    FOR(i,1,n)  si(a[i]);
    FOR(i,1,n-1)
    {
        si(x),si(y);
        g[x].pb(y);
        g[y].pb(x);
    }
    dfs1(1,1);
    REW(vis,0);
    dfs2(0,1);//第一次dfs任选一个节点求出pos，pos肯定是树的一个端点
    REW(vis,0);
    dfs2(0,pos);
    cout<<(deep+1)/2<<endl;
    return 0;
}

也可以用并查集缩点

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define pb push_back
#define lc (d<<1)
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
int n,x,y,a[200008],deep,pos,fa[200008];
vector<int>g[200008],ne[200008];
bool vis[200008];
int fid(int x)
{
    int a,b=x;
    while(b!=fa[b])  b=fa[b];
    while(x!=b)  a=fa[x],fa[x]=b,x=a;
    return b;
}
void dfs2(int d,int x)//两次DFS求树的直径
{
    if(vis[x])  return;
    vis[x]=1;
    if(d>deep)  deep=d,pos=x;
    for(int i=0;i<ne[x].size();i++)
    {
        dfs2(d+1,ne[x][i]);
    }
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    cin>>n;
    FOR(i,1,n)  si(a[i]),fa[i]=i;
    FOR(i,1,n-1)
    {
        si(x),si(y);
        if(a[fid(x)]==a[fid(y)])  fa[fid(x)]=fid(y);
        g[x].pb(y);
        g[y].pb(x);
    }
    FOR(i,1,n)
    {
        if(fid(i)==i)  pos=i;
        for(int j=0;j<g[i].size();j++)
        {
            if(a[fid(i)]!=a[fid(g[i][j])])
            {
                ne[fid(i)].pb(fid(g[i][j]));
                ne[fid(g[i][j])].pb(fid(i));
            }
        }
    }
    REW(vis,0);
    dfs2(0,pos);//第一次dfs任选一个节点求出pos，pos肯定是树的一个端点
    REW(vis,0);
    dfs2(0,pos);
    cout<<(deep+1)/2<<endl;
    return 0;
}