P2921 [USACO08DEC]Trick or Treat on the Farm G

题目链接：

P2921 [USACO08DEC]Trick or Treat on the Farm G - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

解题思路：

看到走回原处即存在环，那么就用tarjan算法，这题首先要判断自环

分为两种情况：

ans[i]为第i头牛走过的隔间数目，id[i]表示点 i 在id[i]这个连通块中，Size[id[i]]表示点 i 所在的连通块的隔间数目

1：若点 i 在环中：则ans[i] = Size[id[i]]

2：若点 i 不在环中：则ans[i] = 从点 i 走到最近的一个环的步数 + 环中点的隔间数目。

#include <stack>
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 100010, M = 100010;

int n;
int id[N];
int ans[N];
int Next[N];
stack<int> stk;
bool in_stk[N];
int Size[N], scc_cnt;
int h[N], e[M], ne[M], idx;
int low[N], dfn[N], timestamp;

void add(int a, int b)//邻接表模板
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx;
    idx ++ ;
}

void tarjan(int u)//tarjan模板
{
    stk.push(u), in_stk[u] = true;
    low[u] = dfn[u] = ++ timestamp;
    
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (!dfn[j])
        {
            tarjan(j);
            low[u] = min(low[u], low[j]);
        }
        else if (in_stk[j]) low[u] = min(low[u], dfn[j]);
    }
    
    if (dfn[u] == low[u])
    {
        int y;
        scc_cnt ++ ;
        do
        {
            y = stk.top();
            stk.pop();
            in_stk[y] = false;
            id[y] = scc_cnt;
            Size[scc_cnt] ++ ;
        } while (y != u);
    }
}

void search(int u, int ne, int step)
{
    if (ans[ne])//若点 u 所连接的下一个点是环
    {
        ans[u] = ans[ne] + step;//则ans[u] = 环中隔间的个数 + 从u走到环的步数
        return ;
    }
    
    search(u, Next[ne], step + 1);//不存在环继续走，步数加1
}

int main()
{
    cin >> n;
    memset(h, -1, sizeof h);
    for (int i = 1; i <= n; i ++ )
    {
        scanf("%d", &Next[i]);
        add(i, Next[i]);
        if (Next[i] == i) ans[i] = 1;//说明存在自环，则点 i 的隔间数为1
    }
    
    for (int i = 1; i <= n; i ++ )
        if (!dfn[i])
            tarjan(i);
    
    for (int i = 1; i <= n; i ++ )//判断点 i 是否在环中
        if (Size[id[i]] > 1) ans[i] = Size[id[i]];//说明点 i 在一个环中，直接赋值即可
    
    for (int i = 1; i <= n; i ++ )
        if (!ans[i]) search(i, Next[i], 1);//说明点 i 不在环中
    
    for (int i = 1; i <= n; i ++ ) printf("%d\n", ans[i]);
    
    return 0;
}