CodeForces 734 E.Anton and Tree（dfs）

Description
给出一棵树，每个点有黑白两色，对于一个点v操作一次，会把所有到v路径上点全部和v同色的点u染成另一种颜色，问至少需要多少次操作才能使得整棵树同色
Input
第一行一整数n表示点数，之后n个整数c[i]表示第i个点的颜色，0表示白色1表示黑色，之后输入该棵树的n-1条边(1<=n<=2e5)
Output
输出最少操作数
Sample Input
11
0 0 0 1 1 0 1 0 0 1 1
1 2
1 3
2 4
2 5
5 6
5 7
3 8
3 9
3 10
9 11
Sample Output
2
Solution
一遍dfs把连通的一块同样颜色的点缩成一个点构成一棵新树，新树满足任意一个点和其父亲节点颜色不同，以新树直径终点作为根将树有向化之后树变成一个黑白相间的(d+1)/2层的树，d为新树的直径，那么就可以从根节点开始一层层染色，只需d+1)/2次操作即可将整棵树变成同色
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 222222
int n,c[maxn],vis[maxn],ans,pos;
vector<int>g[maxn],gg[maxn]; 
void dfs(int u,int fa)
{
    if(vis[u])return ;
    vis[u]=1;
    if(c[u]!=c[fa])gg[fa].push_back(u),gg[u].push_back(fa),fa=u;
    for(int i=0;i<g[u].size();i++)dfs(g[u][i],fa);
}
void dfs1(int u,int deep)
{
    if(vis[u])return ;
    vis[u]=1;
    if(deep>ans)ans=deep,pos=u;
    for(int i=0;i<gg[u].size();i++)dfs1(gg[u][i],deep+1);
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&c[i]);
            vis[i]=0,g[i].clear(),gg[i].clear();
        }
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            g[u].push_back(v),g[v].push_back(u);
        }
        memset(vis,0,sizeof(vis));
        dfs(1,1);
        memset(vis,0,sizeof(vis));
        ans=0;
        dfs1(1,0); 
        memset(vis,0,sizeof(vis));
        ans=0;
        dfs1(pos,0);
        printf("%d\n",(ans+1)/2);
    }
    return 0;
}