L2-031 深入虎穴 (25分)
著名的王牌间谍 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
思路分析:
我们采用并查集的思路data[ ]保存根结点和子结点的关系,rank[ ]保存的是距离根的距离。
此题的树的结构天然的就只有一个根,没有合并的操作,美滋滋,幸福!
思路:
遍历所有的结点,找出他么各自到根结点的距离,就是从每一个结点一直向上找,找到它的根,显然我们可以意识到改进的地方,比如:我们从A结点开始往上找到它的根,那么如果C是A的后代,那么我们通过C来找到A的时候,就直接把A的rank[A]拿过来就行了,没必要再走一遍。
#include <iostream>
#include<stdlib.h>
#include<cstring>
#include<stdio.h>
using namespace std;
//mh是核心
int mh(int rank[],int data[],int index){
//如果这个,结点的rank已经有了,直接return即可
if(rank[index] != 1)
return rank[index];
//向上找它的上一个结点的rank
if(data[index] != -1){
rank[index] = mh(rank,data,data[index])+1;
return rank[index];
}
else{
//根结点的rank是1
return 1;
}
}
int main() {
int N,k;
cin>>N;
int data[N+1];
int rank[N+1];
for(int i=0;i<=N;i++){
data[i]=-1;
rank[i]=1;
}
int total = 1,index = 0;
//i从1开始
//int i;
for(int i=1;i<=N;++i){
cin>>k;
if(k==0)continue;
for(int j=0;j<k;++j){
int temp;
cin>>temp;
data[temp] = i;
}
}
for(int i=1;i<=N;i++){
mh(rank,data,i);
}
int max = 0;
int sign;
for(int i=1;i<=N;i++){
if(max<rank[i])
{
max = rank[i];
sign = i;
}
}
cout<<sign;
return 0;
}