Codeforces Round #363 (Div. 1) B. Fix a Tree 拆环+合并树

传送门

思路：

这里的树如果没有形成环的话，用并查集就可以搞定了，可是有环的话并查集会无限递归，怎么办呢？

解环，这里用的是一个vis数组，每次从某个节点不停向上找，找到已经访问过的就停止（这里既可以判环也可以判是否连通），用index来表示每一次访问，如果之间已经访问过，但这次同样在这里终止循环，那么说明这是一个共同的树根，无须处理，如果这是第一次访问到这个根或者环，那么我们统统把这个根或者环的终点（因为采用了循环，所以会有几个终点）指向我们标好的s树根，并且次数加+1.

GG

#include <cstdio>
#include<iostream>
using namespace std;
const int maxn = 200200;
int f[maxn], vis[maxn], n, s, cnt, idx;
int Find(int x)
{
vis[x] = ++ idx;
while (!vis[ f[x] ])
{
x = f[x];
vis[x] = idx;
}
if (vis[ f[x] ] == idx)
{
if (s == 0)//s==0表明全是环，这时我们要从这些环里再确定一个根，
s = x;
if (f[x] != s)
{
f[x] = s;
cnt ++;
}
}
}
int main()
{
scanf("%d", &n);
for (int i = 1; i <= n; i ++)
scanf("%d" , f + i);
for (int i = 1; i <= n; i ++)
if (i == f[i])
s = i;
for (int i = 1; i <= n; i ++)
if (!vis[i])
Find(i);
printf("%d\n", cnt);
for (int i = 1; i <= n; i ++)
printf("%d ", f[i]);
return 0;
}