洛谷 P2661：信息传递 ← 树形DP+拓扑排序+dfs

【题目来源】
https://www.luogu.com.cn/problem/P2661

【题目描述】
有 n 个同学（编号为 1 到 n）正在玩一个信息传递的游戏。在游戏里每人都有一个固定的信息传递对象，其中，编号为 i 的同学的信息传递对象是编号为 Ti 的同学。
游戏开始时，每人都只知道自己的生日。之后每一轮中，所有人会同时将自己当前所知的生日信息告诉各自的信息传递对象（注意：可能有人可以从若干人那里获取信息，但是每人只会把信息告诉一个人，即自己的信息传递对象）。当有人从别人口中得知自己的生日时，游戏结束。请问该游戏一共可以进行几轮？

【输入格式】
输入共 2 行。
第一行包含 1 个正整数 n，表示 n 个人。
第二行包含 n 个用空格隔开的正整数 T1,T2 ,⋯,Tn，其中第 i 个整数 Ti 表示编号为 i 的同学的信息传递对象是编号为 Ti 的同学，Ti≤n 且 Ti≠i。

【输出格式】
共一行一个整数，表示游戏一共可以进行多少轮。

【输入样例】
5
2 4 2 3 1

【输出样例】
3

【说明/提示】
样例解释：游戏的流程如图所示。

当进行完第 3 轮游戏后，4 号玩家会听到 2 号玩家告诉他自己的生日，所以答案为 3。当然，第 3 轮游戏后，2 号玩家、3 号玩家都能从自己的消息来源得知自己的生日，同样符合游戏结束的条件。
对于 30% 的数据，n≤200；
对于 60% 的数据，n≤2500；
对于 100% 的数据，n≤2×10^5。

【算法分析】
● 链式前向星：https://blog.csdn.net/hnjzsyjyj/article/details/139369904

【算法分析】

/* 链式前向星存图
val[idx]：存储编号为 idx 的边的值
e[idx]：存储编号为 idx 的结点的值
ne[idx]：存储编号为 idx 的结点指向的结点的编号
h[a]：存储头结点 a 指向的结点的编号
*/

#include <bits/stdc++.h>
using namespace std;

const int inf=0x3f3f3f3f;
const int maxn=2e5+5;
const int maxm=maxn<<1;
int e[maxm],ne[maxm],h[maxn],idx;
int stk[maxn];
int ind[maxn]; //in-degree
bool st[maxn];
int ans=inf;
int n,top;

void add(int a,int b) {
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void topo() {
    for(int i=1; i<=n; i++)
        for(int j=h[i]; ~j; j=ne[j]) {
            int k=e[j];
            ind[k]++;
        }

    for(int i=1; i<=n; i++) {
        if(ind[i]==0) stk[++top]=i;
    }

    int t;
    while(top) {
        t=stk[top--];
        st[t]=1;
        for(int i=h[t]; ~i; i=ne[i]) {
            int j=e[i];
            if(--ind[j]==0) stk[++top]=j;
        }
    }
}

void dfs(int u,int len) {
    st[u]=1;
    for(int i=h[u]; ~i; i=ne[i]) {
        int j=e[i];
        if(!st[j]) {
            len++;
            dfs(j,len);
        } else {
            ans=min(ans,len);
            return;
        }
    }
}

int main() {
    memset(h,-1,sizeof(h));
    scanf("%d",&n);

    int t;
    for(int i=1; i<=n; i++) {
        scanf("%d",&t);
        add(i,t);
    }

    topo();
    for(int i=1; i<=n; i++) {
        if(!st[i]) dfs(i,1);
    }

    printf("%d\n",ans);
}

/*
in:
5
2 4 2 3 1

out:
3
*/

【参考文献】
https://www.luogu.com.cn/problem/solution/P2661
https://blog.csdn.net/hnjzsyjyj/article/details/139627833
https://blog.csdn.net/xiji333/article/details/102164195