题目描述
有 n 个同学(编号为 1 到 n )正在玩一个信息传递的游戏。在游戏里每人都有一个固定的信息传递对象,其中,编号为 i 的同学的信息传递对象是编号为 Ti的同学。
游戏开始时,每人都只知道自己的生日。之后每一轮中,所有人会同时将自己当前所知的生日信息告诉各自的信息传递对象(注意:可能有人可以从若干人那里获取信息, 但是每人只会把信息告诉一个人,即自己的信息传递对象)。当有人从别人口中得知自 己的生日时,游戏结束。请问该游戏一共可以进行几轮?
输入格式
共2行。
第1行包含1个正整数 n ,表示 n 个人。
第2行包含 n 个用空格隔开的正整数 T1,T2…Tn,其中第 i 个整数 Ti表示编号为 i 的同学的信息传递对象是编号为 Ti的同学 Ti≤n 且 Ti!=i 。
输出格式
1个整数,表示游戏一共可以进行多少轮。
输入输出样例
输入
5
2 4 2 3 1
输出
3
说明/提示
样例1解释
游戏的流程如图所示。当进行完第3 轮游戏后, 4号玩家会听到 2 号玩家告诉他自己的生日,所以答案为 3。当然,第 3 轮游戏后,2号玩家、 3 号玩家都能从自己的消息来源得知自己的生日,同样符合游戏结束的条件。
对于 30%的数据, n≤200;
对于 60%的数据, n≤2500;
对于100%的数据, n≤200000。
不妨将每个同学看作一个结点,每个结点之间的有向边就是同学之间的生日信息的传递,当某个同学从另一个同学那知道了自己的生日信息的时刻则形成了最小环。游戏轮数就是最小环的有向边的条数。
运用维护额外信息的并查集可以实现图论求最小环。
为了方便求得最小环的长度,我们可以构造维护到结点祖宗结点距离的并查集。
题目中给出了1~n号结点所对应的信息传输对象,以此可以构造各个集合(get_ans)且更新dis数组,注意dis数组代表的含义。
也就是说此时已经建好了个图,之后信息传递就按照构造的图的这些路径。
当游戏结束,最小环的长度,即有向边的数量就是进行的游戏轮数。
(最小环形成于当两个结点的祖宗节点相同时,其长度为两个结点到其祖宗结点距离的和加上1,画图很好理解,就是最小环包含的结点数)
有的也管祖宗节点叫做父节点,是一个意思
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 2e5 + 10;
int n;
int p[N];//p[]存储每个点的祖宗节点
int dis[N];//dis[x]存储x到祖宗节点的距离
int ans = 0x3f3f3f3f;
int find(int x)
{
if (p[x] != x)
{
int t = find(p[x]);
dis[x] += dis[p[x]];//**这里的dis[p[x]]是p[x]到它的父节点的距离,原本dis[x]也是存储x到父节点的距离,然后p[x]变成了根节点,dis[x]才成了x到根节点的距离
p[x] = t;
}
return p[x];//返回x的祖宗节点
}
void get_ans(int a, int b)
{
//注意a,b是任意两结点,x,y分别是a,b所属根节点
int x = find(a), y = find(b);
if (x != y)//如果a,b不属于同一个集合,则将a所属集合合并到b所属集合,并更新dis[a]
{
p[x] = y;
dis[a] = dis[b] + 1;
}
else//如果a,b同属一个集合,说明已成环,更新最小环的长度
{
ans = min(ans, dis[a] + dis[b] + 1);
}
}
int main()
{
cin.tie(0), ios::sync_with_stdio(false);
cin >> n;
for (int i = 1; i <= n; i++)
p[i] = i;
for (int i = 1; i <= n; i++)
{
int a;
cin >> a;
get_ans(i, a);
}
cout << ans << endl;
return 0;
}