PAT L2-031 深入虎穴(求树的深度)

题目描述

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

这题很明显就是一个树,你只需要先找出来树的根节点,然后根据已知的关系建树,最后遍历一下,找出离根最远的节点。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+100;
const int inf=0x3f3f3f3f;
struct node{
    int fm,bf;
    vector<int>hz;
}a[maxn];
int pre[maxn],ans[maxn],mx,k;
int unionserch(int root)//寻找根节点
{
    while(root!=pre[root])
    {

        root=pre[root];
    }
    return root;
}
void bfs(int i,int n)//建树
{
      a[i].bf=n;
    if(n>mx)mx=n;
    for(int j=0;j<a[i].hz.size();j++)
    {
        bfs(a[i].hz[j],n+1);
    }
}
int main()
{
    int n1;
    cin>>n1;
    for(int i=1;i<=n1;i++)
        pre[i]=i,a[i].fm=a[i].bf=-2;
    for(int i=1;i<=n1;i++)
    {
        int N,x;
        cin>>N;
        for(int j=1;j<=N;j++)
        {
            cin>>x;
            a[i].hz.push_back(x);
            pre[x]=i;
        }
    }
    int x=unionserch(1);

    bfs(x,1);
    int pos;
    for(int i=1;i<=n1;i++)
    {
        if(a[i].bf==mx)
        {
            pos=i;
        }
    }
   cout<<pos<<endl;
    system("pause");
    return 0;

}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章