深入虎穴(dfs)

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

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

输入格式:

输入首先在一行中给出正整数 N(<10​5​​),是门的数量。最后 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

思路:首先要找到入口,很显然,这里每条边实际上是有向的,每扇门指向其后面的门,所以入口就是没有被任何一扇门指向的那扇门(即:n行中没有出现的编号,即为入口编号) ,然后dfs找到距离最远的点即可(dfs时记录深度len,更新最大深度maxlen的同时更新距离最远的点的编号point即可),由于这个题实际上是一个不存在平行边的有向根树,入口即为树根,所以dfs时可以不用vis数组标记遍历过的点(dfs中vis数组标记多数用于避免访问重边(而这里没有重边,所以不用)),一开始用的vis数组+depth数组(记录每个点深度),后来发现这样可以做但完全没必要,只需要在dfs中加一个参数记录当前深度即可。

初始AC代码:

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+5;
vector<int>v[maxn];
int n;

int maxlen=-1,point,vis[maxn],depth[maxn];
void dfs(int x)
{
    vis[x]=1;
    for(int i=0;i<v[x].size();i++){
        int p=v[x][i];
        if(!vis[p]){
            vis[p]=1;
            depth[p]=depth[x]+1;
            if(depth[p]>maxlen){
                maxlen=depth[p];
                point=p;
            }
            dfs(p);
            vis[p]=0;
        }
    }
}

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++){
        int k;
        cin>>k;
        while(k--){
            int x;
            cin>>x;
            v[i].push_back(x);
            vis[x]=1;
        }
    }
    if(n==1){
        cout<<1<<endl;
        return 0;
    }
    int start;
    for(int i=1;i<=n;i++){
        if(!vis[i]){
            start=i;
            break;
        }
    }
    memset(vis,0,sizeof(vis));
    dfs(start);
    cout<<point<<endl;
    return 0;
}

简化后的AC代码:

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+5;
vector<int>v[maxn];
int n,vis[maxn];

int maxlen=-1,point;
void dfs(int x,int len)
{
    len++;
    if(len>maxlen){
        maxlen=len;
        point=x;
    }
    for(int i=0;i<v[x].size();i++){
        dfs(v[x][i],len);
    }
}

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++){
        int k;
        cin>>k;
        while(k--){
            int x;
            cin>>x;
            v[i].push_back(x);
            vis[x]=1;
        }
    }
    int start;
    for(int i=1;i<=n;i++){
        if(!vis[i]){
            start=i;
            break;
        }
    }
    dfs(start,0);
    cout<<point<<endl;
    return 0;
}
发布了176 篇原创文章 · 获赞 9 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/Mr_Kingk/article/details/104186019