https://www.luogu.com.cn/problem/P2661
题目描述:
有
n
n
n个同学(编号为
1
1
1到
n
n
n)正在玩一个信息传递的游戏。在游戏里每人都有一个固定的信息传递对象,其中,编号为
i
i
i的同学的信息传递对象是编号为
T
i
T_i
Ti的同学。游戏开始时,每人都只知道自己的生日。之后每一轮中,所有人会同时将自己当前所知的生日信息告诉各自的信息传递对象(注意:可能有人可以从若干人那里获取信息, 但是每人只会把信息告诉一个人,即自己的信息传递对象)。当有人从别人口中得知自 己的生日时,游戏结束。请问该游戏一共可以进行几轮?
输入格式:
共
2
2
2行。
第
1
1
1行包含
1
1
1个正整数
n
n
n ,表示
n
n
n个人。
第
2
2
2行包含
n
n
n个用空格隔开的正整数
T
1
,
T
2
,
⋯
⋯
,
T
n
T_1,T_2,\cdots\cdots,T_n
T1,T2,⋯⋯,Tn,其中第
i
i
i个整数
T
i
T_i
Ti表示编号为
i
i
i的同学的信息传递对象是编号为
T
i
T_i
Ti的同学,
T
i
≤
n
T_i \leq n
Ti≤n且
T
i
≠
i
T_i \neq i
Ti=i 。
输出格式:
1
1
1个整数,表示游戏一共可以进行多少轮。
对于
30
%
30\%
30%的数据,
n
≤
200
n ≤ 200
n≤200;
对于
60
%
60\%
60%的数据,
n
≤
2500
n ≤ 2500
n≤2500;
对于
100
%
100\%
100%的数据,
n
≤
200000
n ≤ 200000
n≤200000。
本质上是要在一个无权有向图上求最短环,并且该图的每个顶点最多只有一个出度,可以用带权并查集来做。设 d [ x ] d[x] d[x]是 x x x沿着出边走多少步能走到 p [ x ] p[x] p[x]。每次找某个点 u u u的祖宗的时候,如果 u = p [ u ] u=p[u] u=p[u],则 u u u自己就是树根,直接返回;否则先递归找 p [ u ] p[u] p[u]的祖宗,并且设递归完成之后, p [ u ] p[u] p[u]及其上方的点的 d d d值都算对了,接着由于要将 u u u连到祖宗上,所以先更新 d [ u ] d[u] d[u]为 d [ u ] + d [ p [ u ] ] d[u]+d[p[u]] d[u]+d[p[u]],然后连到祖宗上。
设 a [ u ] a[u] a[u]是 u u u指向的点的编号,遍历 a a a数组,遍历到 u u u的时候,看一下 a [ u ] a[u] a[u]的祖宗是谁,容易知道 a [ u ] a[u] a[u]能走到 u u u等价于 a [ u ] a[u] a[u]在并查集里的祖宗一定是 u u u(因为每个点的出边只有一条,而 u u u在并查集里的出边暂时还是指向自己的),所以,如果 a [ u ] a[u] a[u]的祖宗是 u u u,则说明找到了一条环,其长度即为 d [ a [ u ] ] + 1 d[a[u]]+1 d[a[u]]+1,用该值更新答案;否则的话,说明没找到环,则令 p [ u ] = a [ u ] , d [ u ] = 1 p[u]=a[u], d[u]=1 p[u]=a[u],d[u]=1。代码如下:
#include
using namespace std;
const int N = 2e5 + 10;
int n, p[N], d[N], res = 2e9;
int find(int x) {
if (p[x] != x) {
int px = find(p[x]);
d[x] += d[p[x]];
p[x] = px;
}
return p[x];
}
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; i++) p[i] = i;
for (int i = 1; i <= n; i++) {
int x;
scanf("%d", &x);
if (find(x) == i) res = min(res, d[x] + 1);
else p[i] = x, d[i] = 1;
}
printf("%d\n", res);
}
时间复杂度 O ( n log ∗ n ) O(n\log^*n) O(nlog∗n),空间 O ( n ) O(n) O(n)。