洛谷P2661 信息传递(带权并查集求有向图最小环)

时间:2022-10-30 17:04:54

题目描述

有n个同学(编号为1到n)正在玩一个信息传递的游戏。在游戏里每人都有一个固定的信息传递对象,其中,编号为i的同学的信息传递对象是编号为Ti同学。

游戏开始时,每人都只知道自己的生日。之后每一轮中,所有人会同时将自己当前所知的生日信息告诉各自的信息传递对象(注意:可能有人可以从若干人那里获取信息,但是每人只会把信息告诉一个人,即自己的信息传递对象)。当有人从别人口中得知自己的生日时,游戏结束。请问该游戏一共可以进行几轮?

输入输出格式

输入格式:

输入共2行。

第1行包含1个正整数n表示n个人。

第2行包含n个用空格隔开的正整数T1,T2,……,Tn其中第i个整数Ti示编号为i

的同学的信息传递对象是编号为Ti的同学,Ti≤n且Ti≠i

数据保证游戏一定会结束。

输出格式:

输出共 1 行,包含 1 个整数,表示游戏一共可以进行多少轮。

输入输出样例

输入样例#1:  复制
52 4 2 3 1
输出样例#1:  复制
3

说明

样例1解释

洛谷P2661 信息传递(带权并查集求有向图最小环)

游戏的流程如图所示。当进行完第 3 轮游戏后, 4 号玩家会听到 2 号玩家告诉他自

己的生日,所以答案为 3。当然,第 3 轮游戏后, 2 号玩家、 3 号玩家都能从自己的消息

来源得知自己的生日,同样符合游戏结束的条件。

对于 30%的数据, n ≤ 200;

对于 60%的数据, n ≤ 2500;

对于 100%的数据, n ≤ 200000。

用带权并查集求最小环,每次建边u-->v,长度为1,然后一直合并,当u和v在同一集合,此时成环,环的大小等于v到根节点的长度+1(因为r[x]求的是x到其父节点的权值,我的代码里默认使父节点为v,f[tx]=ty),然后刷新最小值即可

洛谷P2661 信息传递(带权并查集求有向图最小环)比如这个例子

加入第一条边,x1的祖先节点变为x2且x1到x2的距离为1,即f[x1]=x2,r[x1]=1

加入第二条边,x2的祖先节点变为x3且x2到x3的距离为2,即f[x2]=x3,r[x2]=1

加入第三条边,x3的祖先节点和x1的祖先节点相同,此时成环,然后经过路径压缩f[x1]=x3,r[x1]=2

此时x1到他的祖先节点x3的权值为2,在加上第三条边,即2+1=3,环长度为3

#include<bits/stdc++.h>#define mem(a,b) memset(a,b,sizeof(a))#define wtf printf("wtf\n");using namespace std;typedef long long ll;const int N=2e5+20;int a[N],f[N],r[N],ans=0x3f3f3f3f;int getf(int x){    if(f[x]==x)return x;    int t=getf(f[x]);    r[x]+=r[f[x]];    f[x]=t;    return f[x];}void mix(int x,int y){    int tx=getf(x);    int ty=getf(y);    if(tx!=ty)    {        f[tx]=ty;        r[tx]=r[y]-r[x]+1;    }    else    {        ans=min(ans,r[y]+1);    }}int main(){    int n;    scanf("%d",&n);    for(int i=1;i<=n;i++)    {        f[i]=i;        scanf("%d",&a[i]);    }    for(int i=1;i<=n;i++)        mix(i,a[i]);    printf("%d\n",ans);}