L2-031 深入虎穴

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

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

输入格式：

输入首先在一行中给出正整数 N（<105），是门的数量。最后 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

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

一开始也打算用dfs或者bfs但是想想快算了，直接用类似并查集那个查最长路径，期间还能标记每个点的最长路，所以只需要把叶子结点存一下一搜就行了。

#include<bits/stdc++.h>
using namespace std;
int N, m = 1, sum = 0;
const int M = 1e5 + 5;
int a[M] = { 0 };
int dis[M] = { 0 };
int find(int x)
{	
	if (dis[x] != 0) return dis[x];
	if (x == a[x]) 	return 0;	
	return dis[x] = 1 + find(a[x]);//存一下已知的路径长度
}
int main()
{
	cin >> N;
	vector<int> v;
	for (int i = 1; i <= N; i++)
	{
		int k,b;
		cin >> k;
		if (k == 0) v.push_back(i);//叶子k是0
		for (int o = 0; o < k; o++)
		{
			cin >> b;
			a[b] = i;
		}
	}
	for (auto i:v)
	{
		int f = find(i);
		if (sum < f)
		{
			sum = f;
			m = i;
		}
	}
	cout << m;
	return 0;
}