pta—深入虎穴

记录数据结构pta作业

题目

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

思路

1. 注意，编号为1的门不一定是入度为0的起点门
2. 首先，找到入度为0的起点门
3. 用DFS算法找到最深的深度

代码

#include<iostream>
#include<array>
#include<queue>
using namespace std;
typedef struct Node{
    int data;
    int* index;
}Node,*pNode;
void findmax(pNode door,int index,int &level,int &max,int &num){
    if(door[index-1].data!=0){
        int levelnext = level+1;
        if(levelnext>max){
            max = levelnext;
            num = door[index-1].index[0];
        }
        for(int i = 0; i < door[index-1].data; i++){
            int e = door[index-1].index[i];
            findmax(door,e,levelnext,max,num);
        }
    }
}
int main(){
    int N,n,next;
    cin>>N;
    pNode door = new Node[N];
    int* start = new int[N];
    for(int i = 0; i < N; i++){
        cin >> n;
        door[i].data = n;
        door[i].index = new int[n];
        for(int j = 0; j < n; j++){
            cin >> next;
            door[i].index[j] = next;
            start[next-1]++;
        }
    }    
    int level = 1;
    int max = 1;
    int num=0;
    int startnum = 0;
    for(int i = 0; i < N; i++){
        if(start[i]==0){
           num =startnum =  i+1;
        }
    }

    findmax(door,startnum,level,max,num);
    cout<<num;

    system("pause");
    return 0;
}