丢手绢

链接：https://www.nowcoder.com/questionTerminal/83e03afc8b6745c893362febebb076bb
来源：牛客网
 

有n 个老铁（编号为 1 到n）正在玩丢手绢。在游戏里每人会把当前手里的手绢丢给一个固定的人，编号为Ti。 游戏开始时，每人手里有自己的手绢。之后每一轮中，所有人会同时将自己当前手里的手绢全部丢给接收的对象。当有人重新拿到自己的手绢时，游戏结束。

那么游戏几轮会结束呢？

 

输入描述:

输入共 2 行。
第1 行包含 1 个正整数 n ，表示 n 个人。（n<=200000）
第2 行包含n 个用空格隔开的正整数T1,T2,… ,Tn，其中第Ti个整数表示编号为i 的同学会将手绢丢给编号为 Ti 的同学，Ti ≤n 且Ti ≠i。
保证游戏一定会结束。

 

输出描述:

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

示例1

输入

5
2 4 2 3 1

输出

3

这道题关键在于找一个最小环，用dfs。难点在于记录上一轮dfs（不同的开始节点）的点的记录，如果不记录的话，就不能知道是不是本轮的，导致时间戳混用。记录一开始我使用memset一遍每个点的时间戳为-1，超时，只能完成90%，要使用更高效的方法，不要memset，那就使用visited的取值为0,1,2,2表示已经完成，再dfs后对visited再加一。

#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
int n;
int arr[200005];
int visited[200005];
int d[200005];
int ans;
  
void dfs(int pos,int depth){
    if(visited[pos]==2){
        return;
    }
    if(visited[pos]==1){
        ans=min(ans,depth-d[pos]);
        return;
    }
    visited[pos]++;
    d[pos]=depth;
    dfs(arr[pos],depth+1);
    visited[pos]++;
}
     
  
int main(){
    memset(visited,0,sizeof(visited));
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&arr[i]);
    }
      
    ans=n+1;
    for(int i=1;i<=n;i++){
        if(visited[i]==2){
            continue;
        }
        dfs(i,0);       
    }
      
    printf("%d\n",ans);
}