【noip 2015】信息传递

题目描述

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

数据范围：

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

---

这道题其实就是个简单的dfs，然而蒟蒻的我还是没有一次AC。想了一个小技巧，看题解里没有人用，就来总结一下。

这道题打眼看上去可以转换成一张有向有环图，而题目要求的就是最小环。貌似一个dfs就可以解决了。我原本的想法是对于每一个点都进行一遍dfs，但是是O(n^2)的解法，肯定会t。

于是又想了一下，已经dfs过的点就不用dfs了。所以我就开了一个vis数组，把访问过的点打上标记并且不清零，遇到标记就判断能否更新答案。提交了一遍发现WA 了。

本来以为这是一道sb题，结果自己sb了。

这样有一个bug，就是vis标记不一定是这一次dfs打上的，不一定是环。

举个例子，两条路1->3->5->2->……和4->5->7->……
1->3>5>2这条路搜索完了发现没有环，然后返回主程序开始dfs4这个点，发现5已经打上标记了，程序就开始更新答案，但是并不是环。

于是我又开了一个tmpvis数组，表示这个点是不是在这一次被搜索到的。如果是，就表示有环，更新答案。如果不是，就不更新，continue。

代码如下：

#include<cstdio>
#include<iostream>
using namespace std;
struct node{
    int to,next;
}mem[200001];
int head[200001]={0};
int size=0;
int dep[200001]={0};
int ans=1e9;
bool vis[200001]={0};
bool tmpvis[200001]={0};
int read()
{
    int x=0;
    char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x;
}
void add(int from,int to)
{
    size++;
    mem[size].to=to;
    mem[size].next=head[from];
    head[from]=size;
}
void dfs(int x,int depth)
{
    dep[x]=depth;
    for(int i=head[x];i;i=mem[i].next)
    {
        int v=mem[i].to;
        if(!vis[v])
        {
            vis[v]=1;
            tmpvis[v]=1;
            dfs(v,depth+1);
            tmpvis[v]=0;
        }
        else if(tmpvis[v])
        {
            ans=min(ans,depth+1-dep[v]);
            continue;
        }
    }
}
int main()
{
    int n;
    n=read();
    int t;
    for(int i=1;i<=n;i++)
    {
        t=read();
        add(i,t);
    }
    for(int i=1;i<=n;i++)
    {   
        if(!vis[i])
        {
            vis[i]=1;
            tmpvis[i]=1;
            dfs(i,1);
            tmpvis[i]=0;
        }
    }
    printf("%d",ans);
    return 0;
}