L2-031 深入虎穴 (25 分)

本次刷题为了2021年天梯

L2-031 深入虎穴 (25 分)

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

题解

发现自己的代码比之前变得优美
写的也更流畅了
确实比去年天梯的时候有很大进步
加油 崇文

Code

BFS

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
const int n = 1e5 + 10;
int dp[n];
vector<int> v[n];

void bfs(int i)
{
    
    queue<int> q;
    q.push(i);
    int ans = 1;
    while (!q.empty())
    {
    
        auto x = q.front();
        q.pop();
        ans = x;
        for(auto &e : v[x])
            q.push(e);

    }
    cout << ans;
}
int main()
{
    
    int N, M, x;
    cin >> N;
    for (int i = 1; i <= N; i++)
    {
    
        cin >> M;
        while (M--)
            cin >> x, v[i].push_back(x), dp[x]++;
    }
    for (int i = 1; i <= N; i++)
        if (!dp[i])
            bfs(i);
    return 0;
}

DFS

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
const int n = 1e5 + 10;
int dp[n];
vector<int> v[n];
int ans = 1;
int maxt = 0;
void dfs(int k, int level)
{
    
    if (maxt < level)
        ans = k, maxt = level;
    for (auto &e : v[k])
        dfs(e, level + 1);
}
int main()
{
    
    int N, M, x;
    cin >> N;
    for (int i = 1; i <= N; i++)
    {
    
        cin >> M;
        while (M--)
            cin >> x, v[i].push_back(x), dp[x]++;
    }
    for (int i = 1; i <= N; i++)
        if (!dp[i])
            dfs(i, 0);
    cout << ans;
    return 0;
}