天梯赛L2-3 深入虎穴 (25 分)

PTA L2-3 深入虎穴 (25 分)
著名的王牌间谍 007 需要执行一次任务，获取敌方的机密情报。已知情报藏在一个地下迷宫里，迷宫只有一个入口，里面有很多条通路，每条路通向一扇门。每一扇门背后或者是一个房间，或者又有很多条路，同样是每条路通向一扇门…… 他的手里有一张表格，是其他间谍帮他收集到的情报，他们记下了每扇门的编号，以及这扇门背后的每一条通路所到达的门的编号。007 发现不存在两条路通向同一扇门。

内线告诉他，情报就藏在迷宫的最深处。但是这个迷宫太大了，他需要你的帮助 —— 请编程帮他找出距离入口最远的那扇门。

输入格式：
输入首先在一行中给出正整数 N<1e5 ，是门的数量。最后 N 行，第 i 行（1≤i≤N）按以下格式描述编号为 i 的那扇门背后能通向的门：

K D[1] D[2] … D[K]
其中 K 是通道的数量，其后是每扇门的编号。

输出格式：
在一行中输出距离入口最远的那扇门的编号。题目保证这样的结果是唯一的。

输入样例：
13
3 2 3 4
2 5 6
1 7
1 8
1 9
0
2 11 10
1 13
0
0
1 12
0
0
输出样例：
12
开始用并查集没做出来，后面发现这题最开始的门就一个，所以计算每个门前面有几个门，找最大值就行了

#include<iostream>
#include<vector>
using namespace std;
const int N = 1e5+5;
vector<vector<int>>p(N);
bool st[100010];
int n;
int cnt[100010];
    int gen;
inline int count(int x){
    if(cnt[x])return cnt[x];//优化，计算过的直接返回
   else if(x==gen){return 1;}//开始的门返回1
    else {
        int l=p[x].size();
        for(int i=0;i<l;++i){
 cnt[x]=count(p[x][i])+1>cnt[x]?count(p[x][i])+1:cnt[x];//因为x门前面可能不止一个门，所以遍历求最大值
        }   
    }
    return cnt[x];
}
int main(){
    int n;
    scanf("%d",&n);
    for(int j(1);j<=n;++j){
        int x;
        scanf("%d",&x);
        for(int i(0);i<x;++i){
            int y;
            scanf("%d",&y);
            p[y].push_back(j);//表示y门在j门后面
            st[y]=1;//标记门
        }
    }
int maxx=0,t;
    for(int i(1);i<=n;++i)if(!st[i]){gen=i;break;}//开始的门前面没有门
    for(int i(1);i<=n;++i){
        cnt[i]=count(i);
       if(maxx<cnt[i]){//找最大值
           t=i;//门的编号
           maxx=cnt[i];
       }
    }
cout<<t;
}