有n个同学(编号为1到n)正在玩一个信息传递的游戏。在游戏里每人都有一个固定的信息传递对象,其中,编号为i的同学的信息传递对象是编号为Ti同学。
游戏开始时,每人都只知道自己的生日。之后每一轮中,所有人会同时将自己当前所知的生日信息告诉各自的信息传递对象(注意:可能有人可以从若干人那里获取信息,但是每人只会把信息告诉一个人,即自己的信息传递对象)。当有人从别人口中得知自己的生日时,游戏结束。请问该游戏一共可以进行几轮?
分析:把题往图上转化,发现是每个点只有一个出边的有向图,要找到里边最小的一个环。最初的思路是先拓扑排序,去掉没用的点,再每个点dfs一遍加上标记就可以了。
但随即发现这个拓扑排序不太好写,一不小心就会超时,毕竟有50万个点。
仔细想想,发现根本不需要先去点,只要每次dfs时打两个标记,一个记录当前点是否搜过以及搜到的时间戳,另一个则记录是否是在同一个点dfs时找到的环。复杂度O(n),
可以过但不知道为什么实际测的时候跑得比预期慢,最慢的一个点有500ms。
#include<iostream>
#include<cstring>
using namespace std;
int n;
int a[500010];
int v[500010];
int u[500010];
int maxx;
int now;
void dfs(int x,int z)
{
if(v[x]!=0)
{
if(u[x]==now)
maxx=min(maxx,z-v[x]);
return ;
}
u[x]=now;
v[x]=z;
dfs(a[x],z+1);
return ;
}
int main()
{
maxx=0x3fffffff;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
for(int i=1;i<=n;i++)
{
now=i;
if(v[i]==0)
{
u[i]=now;
v[i]=1;
dfs(a[i],2);
}
}
cout<<maxx;
return 0;
}