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;

}

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