Codeforces 734E. Anton and Tree By Assassin 缩点+树的最大直径

题意就不多加赘述了，说白了就是每一次可以反转任意连通块儿，这个连通块儿必须满足相连且花色相同，问你最少的反转次数。

思路：一看题目思路其实很清楚了，我们为了缩小数据规模，利用缩点将一个连同块儿化成一个点，之后我们一定可以得到一个树，因为原题就是一个树，之后我们只需要找到树的最大直径就行了。

步骤1：缩点

步骤2：求树的最大直径（其实就是两次搜索，因为第一次从任意点出发找到最远的一个点，那么这个点一定是树直径上两端的一个点，第二次再从这个点出发找到最远的点的距离就是树的直径了。）

第一次写缩点，丑了点。。。

#include<bits/stdc++.h>
#define input freopen("input.txt","r",stdin)
using namespace std;
const int maxn = 200002;
vector<int>vec[maxn],change[maxn];
int vis[maxn]={0},color[maxn]={0},belong[maxn]={0};  // belong代表改点归并到第几个缩点
int n,suo_size=0,ans=maxn,fathur=0;

void dfs(int pos,int number){
int i;
vis[pos]=1;
belong[pos]=number;
for(i=0;i<vec[pos].size();i++){
if(vis[vec[pos][i]]==0&&color[vec[pos][i]]==color[pos]){
dfs(vec[pos][i],number);
}
}
}
void DFS(int pos){
int i;
vis[pos]=1;
for(i=0;i<vec[pos].size();i++){
if(vis[vec[pos][i]]==0&&color[vec[pos][i]]==color[pos]){
DFS(vec[pos][i]);
}
else if(vis[vec[pos][i]]==0&&color[vec[pos][i]]!=color[pos]){
change[belong[vec[pos][i]]].push_back(belong[pos]);
change[belong[pos]].push_back(belong[vec[pos][i]]);
}
}
}
void dps(int pos,int length){
int i,flag=1;
vis[pos]=1;
for(i=0;i<change[pos].size();i++){
if(vis[change[pos][i]]==0){
dps(change[pos][i],length+1);
flag=0;
}
}
if(flag==1){
if(ans<length){
ans=length;
fathur=pos;
}
}
return ;
}
int main(){
int i,j,l,r;
while(scanf("%d",&n)!=EOF){
for(i=1;i<=n;i++) scanf("%d",&color[i]);
for(i=1;i<=n-1;i++){
scanf("%d%d",&l,&r);
vec[l].push_back(r); vec[r].push_back(l);
}
memset(vis,0,sizeof(vis));  //其实大多ACM写成一次处理的，我喜欢加上EOF操作，所以vis要经常初始化，就慢了

for(suo_size=0,i=1;i<=n;i++){  //这次dfs将连通块儿缩到一个点
if(vis[i]==0){
suo_size++;
dfs(i,suo_size);
}
}

memset(vis,0,sizeof(vis));
for(i=1;i<=n;i++){              //这次DFS将缩点后树连接起来
if(vis[i]==0) DFS(i);
}
memset(vis,0,sizeof(vis));

ans=0; fathur=0;            //求树的直径
dps(1,0);
ans=0;                      //第一次的ans求出来没用。。。
memset(vis,0,sizeof(vis));  //记得初始化 访问数组
dps(fathur,0);

cout<<(ans+1)/2<<endl;      //结果先加1因为有奇偶情况，ans求的是两端最远距离，还要加上自己才是直径
}
return 0;
}