分析: 表示树的一维数组,本质上给出了一个权值 t r [ i ] tr[i] tr[i]和当前下标 i i i的索引关系,因此考虑用并查集合并他们,从而考虑以下两种情况
- 不考虑所有错误的点,合并完是一个森林,即有多个祖先 t r [ i ] = i , p [ i ] = i tr[i] = i , p[i] = i tr[i]=i,p[i]=i ,其中错误的点使得其中一个或多个个森林变成了环或者不是我们假定的根节点,因此找到并改变
- 所有点都能合并并且完美合并成一条链,但是其中的点被连接成了环,导致这个图没有根节点(或者说一些根节点构成了一个简单环)找到环上一个点并假设为根节点即可
如果情况为1,则操作数为连通块个数-1
如果情况为2,则操作数为简单环上根节点个数
剩下细节代码展示
#include<bits/stdc++.h>
using namespace std;
const int N = 2e5 + 7;
int p[N] , tr[N];
bool st[N];
int n;
int find(int x) //并查集函数
{
if (p[x] != x) p[x] = find(p[x]);
return p[x];
}
int main()
{
cin >> n;
for (int i = 1; i <= n; ++ i) cin >> tr[i]; 初始数组
for (int i = 1; i <= n; ++ i) p[i] = i; 初始化
int k = n , root = 0; k是连通块个数
for (int i = 1; i <= n; ++ i)
{
if (find(i) != find(tr[i])) 根节点合并
{
p[find(i)] = find(tr[i]);
k --;
}
}
for (int i = 1; i <= n; ++ i)
{
if (p[i] == i && tr[i] == i) root = i; 这句的意思是这个点就是自身的根节点且没有出错
}
int cnt = 0;
if (root)
{
for (int i = 1; i <= n; ++ i)
{
if (p[i] != i) continue; 先看是不是根节点,不是就跳过
if (tr[i] == i) 如果是根节点且为自身
{
if (i != root) tr[i] = root; 合并他们
}
else tr[i] = root;处理环,即当前这个点出错指向了连通块内部// loop
}
cout << k - 1 << endl;
}
else处理环
{
for (int i = 1; i <= n; ++ i)
{
if (p[i] != i) continue;
cnt ++;
if (!root) root = i; 找到环上一个点就假定为他
tr[i] = root;根节点剩下全部赋值即可
}
cout << cnt << endl;
}
for (int i = 1; i <= n; ++ i) cout << tr[i] << ' ';
return 0;
}