luogu#P2921 USACO08DEC在农场万圣节

题意：

给出 \(N\) 个点，每个点指向一个其它点的，求从每个点开始，第二次访问一个节点时走的步数

解析：

Tarjan

#include<cstdio>
#include<stack>
#define N 100005
using namespace std;
struct Edge{int next,to;}edge[N];
int n,n_e,head[N],f[N];
int dfn[N],low[N],tot,clr[N],col,p[N],out[N];
stack<int> s;
bool vis[N];

int dfs(int x);
void tarjan(int u);
inline void addedge(int from,int to);
void init();
int main()
{
    init();
    
    for(int i=1;i<=n;i++)
        if(!dfn[i])
            tarjan(i);

    for(int i=1;i<=n;i++)
        if(clr[i]!=clr[edge[i].to])
            out[clr[i]]=clr[edge[i].to];
    for(int i=1;i<=n;i++)
        printf("%d\n",dfs(clr[i]));
    return 0;
}
int dfs(int x)
{
    if(!x)
        return 0;
    if(f[x])
        return f[x];
    return f[x]=dfs(out[x])+p[x];
}
void init()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int v;
        scanf("%d",&v);
        addedge(i,v);
    }
}
inline void addedge(int from,int to)
{
    edge[++n_e].next=head[from];
    edge[n_e].to=to;
    head[from]=n_e;
}
void tarjan(int u)
{
    tot++; dfn[u]=tot; low[u]=tot;
    s.push(u); vis[u]=1;
    for(int i=head[u];i;i=edge[i].next) 
    {
        int v=edge[i].to;
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[v],low[u]);
        }
        else if(vis[v])
            low[u]=min(low[u],dfn[v]);
    }
    if(low[u]==dfn[u])
    {
        col++; clr[u]=col; p[col]++; vis[u]=0;
        while(s.top()!=u)
        {
            p[col]++;
            vis[s.top()]=0;
            clr[s.top()]=col;
            s.pop();
        }
        s.pop();
    }
}