团体程序设计天梯赛 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

思路

题目需要我们找最远的门，可以用bfs对门进行搜索，为了提高效率，可以对已经走过的门进行剪枝。但此处并没有采用最优的剪枝，只减了“先走远，再走近”的枝，可以进一步优化剪去“先走近，再走远”的枝，这时候需要加一个数组来记录已经访问过的门后面的深度。

代码

#include <iostream>
#include <cstdio>
#include <string>
#include <algorithm>
#include <queue>
#include <stack>
#include <cstring>
#include <cmath>
#include <set>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
struct node{
	int step;
	int num;
};
vector <int> a[maxn];
int vis[maxn];
int main(){
	ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
    	int k;
    	cin>>k;
    	a[i].push_back(k);
    	for(int j=1;j<=k;j++){
    		int t;
    		cin>>t;
    		a[i].push_back(t);
		}
	}
	queue <node> q;
	int numm,summ=0;
	for(int i=1;i<=n;i++){
		if(!vis[i]){
//			vis[i]=1;	//此处不要标记 
			q.push((node){1,i}); 
			while(!q.empty()){
				node t=q.front();
				q.pop();
				if(t.step>summ){
					summ=t.step;
					numm=t.num;
				}
				int k=a[t.num][0];
				for(int j=1;j<=k;j++){
					int tt=a[t.num][j];
					if(!vis[tt]){
						vis[tt]=1;
						q.push((node){t.step+1,tt});
					}
				}
			}
		}
	}
	cout<<numm;
	return 0;
}