洛谷 P2661 信息传递

题目描述

有 $n$ 个同学（编号为 $1$ 到 $n$ ）正在玩一个信息传递的游戏。在游戏里每人都有一个固定的信息传递对象，其中，编号为 $i$ 的同学的信息传递对象是编号为 TiT_iTi​ 的同学。

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

输入输出格式

输入格式：

输入共2行。 第1行包含1个正整数 $n$ ，表示 $n$ 个人。

第2行包含 $n$ 个用空格隔开的正整数 T1,T2,⋯⋯,TnT_1,T_2,\cdots\cdots,T_nT1​,T2​,⋯⋯,Tn​ ，其中第 $i$ 个整数 TiT_iTi​ 表示编号为 $i$ 的同学的信息传递对象是编号为 TiT_iTi​ 的同学， Ti≤nT_i \leq nTi​≤n 且 Ti≠iT_i \neq iTi​≠i 。

输出格式：

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

输入输出样例

输入样例#1：

5
2 4 2 3 1

输出样例#1：

3

说明

样例1解释

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

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

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

对于 30%的数据， n ≤ 200；

对于 60%的数据， n ≤ 2500；

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

显而易见，按照题目中给的样例解释我们是很难看懂的。

这道题在洛谷中的标签是并查集和图论。

但是我没有用到并查集而是拓扑排序+建图遍历。

为什么我会想到拓扑排序呢？

题目中给出每个同学只能把信息交给一个其他的同学。

所以我们可以根据题意建立一个有向图。

然后找到最小环，即为游戏可以进行的轮数。

所以我们可以先用拓扑排序删去不在环中的点。

然后遍历一遍图找到最小环直接输出就行了。

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int MAXN=200000;
int t[MAXN],n,ans=19260817,dg[MAXN],tot=0;
bool flag=0,visit[MAXN];
void topsort()
{
	int q[MAXN*5];
	int head=1,tail=1;
	for(int i=1;i<=n;i++)
		if(!dg[i])
			q[++tail]=i;
	while(head<=tail)
	{
		int x=q[head++];
		int v=t[x];
		dg[v]--;
		if(!dg[v])
			q[++tail]=v;
	}
}
void dfs(int x,int step)
{
	if(!t[x])
		return ;
	if(flag)
		return ;
	if(visit[x])
	{
		ans=min(ans,step);
		flag=1;
		return ;
	}
	visit[x]=1;
	dfs(t[x],step+1);
}
int main()
{
	int x;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>x;
		dg[x]++;
		t[i]=x;
	}
	topsort();
	for(int i=1;i<=n;i++)
	{
		if(dg[i]>0&&visit[i]==0)
		{flag=0; dfs(i,0);}
	}
	cout<<ans;
	return 0;
}