PTA热身赛第二场补题报告

L2-031 深入虎穴 (25 分)

 1.题意　

　　情报藏在一个地下迷宫里，迷宫只有一个入口，里面有很多条通路，每条路通向一扇门。每一扇门背后或者是一个房间，或者又有很多条路，同样是每条路通向一扇门，不存在两条路通向同一扇门，找出距离入口最远的那扇门。给定门的数量N，接下来 N 行，第 i 行描述编号为 i 的那扇门背后能通向的门。在一行中输出距离入口最远的那扇门的编号。

 2.题解

　　题目没给入口，先把入口找到，因为没有哪扇门会通向入口，所以用vis数组记录每个门是否被通向，没被通向的那扇门就是入口。用vector数组存每扇门通往的门，再用dfs搜到最深点，特判只有一扇门的情况，直接输出1。

3.代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;
int n, ru;
int vis[maxn];
vector<int> mp[maxn];
int mmax = 0;
int ans = 0;
void dfs(int pr, int cnt) {
    if(mp[pr].size() == 0) {
        if(cnt > mmax) {
            ans = pr;
            mmax = cnt;
        }
        return;
    }
    
    for(int i = 0; i < (int)mp[pr].size(); i++) {
        dfs(mp[pr][i], cnt + 1);
    }
}
int main() {
    cin >> n;
    for(int i = 1; i <= n; i++) {
        int k;
        cin >> k;
        int x;
        for(int j = 1; j <= k; j++) {
            cin >> x;
            mp[i].push_back(x);
            vis[x] = 1;
        }
    }
    
    if(n == 1) {
        cout << 1 << endl;
        return 0;
    }
    
    for(int i = 1; i <= n; i++) {
        if(vis[i] == 0) {
            ru = i;
            break;
        }
    } 
    
    dfs(ru, 0);
    
    cout << ans << endl;
    
    return 0;
}