E - Anton and Tree(dfs缩点+dfs找最长直径)

题意：给出n个点，n-1条边构成一棵生成树，每个点都有黑白两种颜色，每次改变颜色都能使同颜色的相邻点改变颜色，求最小改变颜色的次数

分析：因为每次改变颜色都能使同颜色的相邻点改变颜色，所以相邻且颜色相同的点可以缩成一点，重新建图，构成黑白相间的点的树，设树的直径(任意两点的最远距离)为d，因为每次改变颜色都会使它与相邻的点的颜色相同，所以取树的直径上的点操作，共有d+1个点，因为点是黑白相间的，所以最少取(d+1)/2个点，即可使全图改变成相同颜色。刚开始死活想不通这一点，后来发现  一些枝叶上的点都会在直径上的点改变颜色的时候，颜色就会改变，而且枝叶上的点长度不可能超过直径上的点，否则就为直径了。

 

#include <iostream>
#include<vector>
#include<stdio.h>
#define N 200005
using namespace std;

vector<int >g[N],gc[N];  //g是原图，gc是缩点后的图
int visit[N],col[N],comp[N],dp[N]; //dp储存该点到图的任意一点的最远距离，comp是该点属于缩点后的图的哪一点
int ans=0;
void dfs(int u,int c,int rt)//缩点操作
{
    if(visit[u] || col[u]!=c)
        return;
    visit[u]=1;
    comp[u]=rt;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        dfs(v,c,rt);
    }
}
void DFS(int u,int fa)//求直径的长度
{
    int mx=0,mmx=0; //mx是点到图的任意一点的最远距离，mmx是次远距离
    for(int i=0;i<gc[u].size();i++)
    {
        int v=gc[u][i];
        if(v==fa)  //防止他遍历回父节点
            continue;
        DFS(v,u);
        mmx=max(mmx,dp[v]+1);
        if(mmx > mx) swap(mx,mmx);
    }
    dp[u]=mx;
    ans=max(ans,mx+mmx);//图的最远距离等于某个点的最远距离加上次最远距离
}
int main()
{
    int n,u,v;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&col[i]);
    }
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    int cnt=1;
    for(int i=1;i<=n;i++)
    {
        if(!visit[i]) dfs(i,col[i],cnt++);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<g[i].size();j++)
        {
            int v= g[i][j];
            if(comp[v]!=comp[i])
            {
                gc[comp[i]].push_back(comp[v]);
            }
        }
    }
    DFS(1,1);
    printf("%d\n",(ans+1)/2);
}