著名的王牌间谍 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;
}