【NOIP2015】洛谷2661 信息传递

题目描述

有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 个整数，表示游戏一共可以进行多少轮。

在一棵dfs树上的返祖边就代表一个环，记一下每个点的深度就可以知道环的大小。

#include<cstdio>
#include<algorithm>
using namespace std;
const int oo=0x3f3f3f3f;
int to[200010],dep[200010],num[200010],tot,ans=oo;
void dfs(int u,int d)
{
    dep[u]=d;
    num[u]=tot;
    if (dep[to[u]])
    {
        if (num[to[u]]<tot) return;
        ans=min(ans,d-dep[to[u]]+1);
    }
    else dfs(to[u],d+1);
}
int main()
{
    int i,j,k,m,n,p,q,x,y,z;
    scanf("%d",&n);
    for (i=1;i<=n;i++)
      scanf("%d",&to[i]);
    for (i=1;i<=n;i++)
      if (!dep[i])
      {
        tot++;
        dfs(i,1);
      }
    printf("%d\n",ans);
}