Codeforces Round #379 (Div. 2) E

对于一棵树，只有两个颜色，每次可以将一个颜色相同的联通块染成一个颜色，求最少的染色次数。

对于每一个颜色相同的联通块缩点，结果得到一个黑白相间的图，这时找出这个图的直径，方法是先搜到最深的一个点以后，将该点作为根在搜一遍找到深度得到的就是这棵树的直径。最后从中间开始不断向两边变色，因为直径总是最长的，所以两边的分支一定在直径变色完都变为一个颜色了，因此只要求直径上的变色次数就行，结果就是(len + 1) / 2。

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 200005;
int col[maxn];
vector<int> G[maxn];
pair<int, int> dfs(int u, int fa, int deep)
{
    pair<int, int> tmp = make_pair(deep, u);
    for(int i = 0; i < G[u].size(); ++i)
    {
        int v = G[u][i];
        if(v != fa)
        {
            if(col[v] == col[u])
                tmp = max(tmp, dfs(v, u, deep));
            else
                tmp = max(tmp, dfs(v, u, deep + 1));
        }
    }
    return tmp;
}
int main()
{
    int n;
    scanf("%d", &n);
    for(int i = 1; i <= n; ++i)
    {
        scanf("%d", &col[i]);
    }
    for(int i = 0; i < n - 1; ++i)
    {
        int u, v;
        scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    pair<int, int> tmp = dfs(1, -1, 0);
    tmp = dfs(tmp.second, -1, 0);
    printf("%d\n", (tmp.first + 1) / 2);
}