PTA 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

实现思路：

1. 深度优先搜索：

   • 定义一个递归函数 dfs，接受当前节点编号 r 和当前路径长度 l 作为参数。
   • 在递归过程中，对于每个节点：
     • 更新最大路径长度 maxl，并记录对应的节点编号 ans[maxl]。
     • 遍历该节点的所有子节点，对每个子节点调用 dfs 函数，传递子节点编号和更新后的路径长度（增加1）。

2. 输入解析：

   • 读取整数 n，表示树的节点总数。
   • 对于每一个节点 i（从1到 n），读取与其相连的子节点数量 num 和具体的子节点编号 e，并建立父子关系，即把子节点的编号加入父节点的 child 列表中。
   • 在读取子节点的同时，使用布尔数组 v 标记哪些节点有父节点，这样可以确定根节点（即没有父节点的节点）。

关键代码：

void dfs(int r,int l)
{
    if(l>maxl)
    {
        maxl=l;
        ans[maxl]=r;
    }
    for(int i=0;i<node[r].child.size();i++)
        dfs(node[r].child[i],l+1);
}

通过截图：

完整代码：

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
struct Node
{
    vector<int> child;
}node[100001];
int n,num,e,r,maxl=-1,ans[100001];
bool v[100001]={false};
void dfs(int r,int l)
{
    if(l>maxl)
    {
        maxl=l;
        ans[maxl]=r;
    }
    for(int i=0;i<node[r].child.size();i++)
        dfs(node[r].child[i],l+1);
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>num;
        for(int j=0;j<num;j++)
        {
            cin>>e;
            v[e]=true;
            node[i].child.push_back(e);
        }
    }
    for(int i=1;i<=n;i++)
        if(v[i]==false)
            r=i;
    dfs(r,0);
    cout<<ans[maxl]<<endl;
    return 0;
}