洛谷 P2921 [USACO08DEC]在农场万圣节Trick or Treat on the Farm（搜索_图的遍历）

传送门

---

似乎有点强连通的意思？？？
直接搜吧，搜的时候记录第一次访问当前点的时间
如果一个点后继牛棚是自己，则形成自环，答案为1
如果一个点在环中且不是自环，则答案为环的大小=第二次搜到当前点的时间-第一次搜到当前点的时间+1，用当前点的答案去更新环内其他点的答案
如果一个点不在环中，则当前点往前搜一定会搜到一个环，递归时更新答案，为当前点的后继牛棚的答案加一。

Code:

#include<cstdio>
#include<cstdlib>
#include<cstring>

int n,m,t;
int a[100010],f[100010],q[100010];

void dfs(int x)
{
    if(x==a[x])
    {
        f[x]=1;
        return;
    }
    if(f[a[x]])
    {
        f[x]=f[a[x]]+1;
        return;
    }
    else
    {
        q[x]=++t;
        if(q[a[x]])
        {
            f[x]=f[a[x]]=t-q[a[x]]+1;
            m=t-q[a[x]]+1;
            return;
        }
        dfs(a[x]);
        if(f[x]==0 && m!=0) f[x]=m;
        else
        {
            m=0;
            if(f[x]==0) f[x]=f[a[x]]+1;
        }
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    memset(q,0,sizeof(q));
    memset(f,0,sizeof(f));
    for(int i=1;i<=n;i++)
    {
        if(!f[i]) t=0,dfs(i);
        printf("%d\n",f[i]);
    }
}