题目大意:对于给定的一颗树,现在已知这棵树的节点只有 0,1 两种颜色,你可以任选一个节点,并将这个节点所在的颜色相通的联通块颜色全部反转。问,最少需要几次操作可以将这棵树上所有的节点变成用一种颜色。
思路: 每次将一个联通块的颜色反转,比较容易想到每次反转后当前这个联通块的区域都会变的更大。
所以我们首先将这棵树进行缩点。缩点之后的树,每个节点都是原来的一个联通快。而将联通块的颜色反转等价于将这个缩点之后的树上的某个节点的全部相连的节点合并到当前节点。这样我们可以发现,我们每次在树上最长的链上做缩点操作能保证次数最少,最少的次数为
length/2
。
#include <cstdio>
#include <string>
#include<iostream>
#include<vector>
#include <stack>
#include <queue>
#include <map>
#include <cstdlib>
#include<string.h>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef pair<int, int>pii;
typedef pair<ll, ll> pll;
typedef pair<int, ll> pil;
const int MAXN = 300000 + 10;
bool vis[MAXN];
int cot[MAXN];
int col[MAXN];
vector<int>g[MAXN];
vector<int>gg[MAXN];
int tot = 0;
int deep = 0,pos=0;
void dfs1(int x,int fa)
{
if (vis[x])return;
vis[x] = true;
if (col[x] != col[fa])gg[x].push_back(fa), gg[fa].push_back(x),fa=x;
for (int i = 0; i < g[x].size(); i++)dfs1(g[x][i], fa);
}
void dfs2(int x, int dp)
{
if (vis[x])return;
vis[x] = true;
if (dp>deep)deep = dp, pos = x;
for (int i = 0; i < gg[x].size(); i++)dfs2(gg[x][i], dp + 1);
}
int main()
{
memset(col, -1, sizeof col);
memset(cot, 0, sizeof cot);
memset(vis, false, sizeof vis);
deep = 0; pos = 0; tot = 0;
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 x, y;
scanf("%d%d", &x, &y);
g[x].push_back(y);
g[y].push_back(x);
}
dfs1(1, 1);
memset(vis, false, sizeof vis);
dfs2(1, 0);
//deep = 0;
memset(vis, false, sizeof vis);
dfs2(pos, 0);
printf("%d\n", (deep+1)/2);
//system("pause");
}