PTA 探索地道战分数 30 (C++)

地道战是在抗日战争时期，在华北平原上抗日军民利用地道打击日本侵略者的作战方式。地道网是房连房、街连街、村连村的地下工事，如下图所示。

我们在回顾前辈们艰苦卓绝的战争生活的同时，真心钦佩他们的聪明才智。在现在和平发展的年代，对多数人来说，探索地下通道或许只是一种娱乐或者益智的游戏。本实验案例以探索地下通道作为内容。

假设探险队发现了一处废弃的地道，它的通道都是直的，而通道所有交叉点（包括通道的端点）上都有一盏灯和一个开关，如下图所示：

队长决定对其进行探索。为了更方便地勘察地形以及规避掉不必要的风险，探索地道的首先需要将整个地道中的所有灯都点亮，不然可能会在地道中迷失方向或者落入陷阱中。

现在已知探索的起点在 S，每一时刻可以从起点派出任意数量的队员出发去点灯（每个位置可以同时存在多个队员），不过每个队员每一时刻最多只能移动一个单位的距离，现在需要输出所有灯被点亮的顺序，以及判断一下最后迷宫中的灯能否全部被点亮。

输入格式:

输入第一行给出三个正整数，分别表示地下迷宫的节点数N（1<N≤1000，表示通道所有交叉点和端点）、边数M（≤3000，表示通道数）和探索起始节点编号S（节点从1到N编号）。随后的M行对应M条边（通道），每行给出一对正整数，分别是该条边直接连通的两个节点的编号。

输出格式:

若可以点亮所有节点的灯，则输出从S开始的序列，序列中相邻的节点一定有边（通道）；否则虽然不能点亮所有节点的灯，但还是输出点亮部分灯的节点序列，最后输出0，此时表示迷宫不是连通图。

由于遍历的节点序列是不唯一的，为了使得输出具有唯一的结果，我们约定以节点小编号优先的次序访问（点灯）。

输入样例1:

6 8 1
1 2
2 3
3 4
4 5
5 6
6 4
3 6
1 5

输出样例1:

1 2 5 3 4 6

输入样例2:

6 6 6
1 2
1 3
2 3
5 4
6 5
6 4

输出样例2:

6 4 5 0

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

栈限制

8192 KB

#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define int long long
#define PII pair<int,int>
#define fi first
#define pb push_back
#define sec second
#define endl '\n'
#define INF 0x3f3f3f3f
const int N=2e5+10;
const int M=1e6+10;
int n,m,k,T;
int vis[N];
set<int>g[N];
//利用set容器在初始存边时就进行排序，遍历的时候自然就是序号小的边在前面 
void bfs(int rt){
	queue<int>q;
	q.push(rt);
	vector<int>v;
	while(!q.empty()){
		int x = q.front();
		q.pop();
		if(vis[x])continue;
		vis[x]=1;
		v.pb(x);
		for(auto i : g[x]){
			if(!vis[i]){
				q.push(i);
//就是正常bfs跑全图然后存储 
			}
		}
	}
	for(int i=0;i<v.size();i++){
		if(i!=0)cout << " " ;
		cout << v[i];
	}
//输出 
	if(v.size()!=n){
		cout << ' ' << 0 << endl;
//如果点的数量对应不上就输出0 
	}else{
		cout << endl;
	}
}
signed main(){
	cin >> n >> m >> k ;
	for(int i=1;i<=m;i++){
		int a , b ;cin >> a >> b ;
		g[a].insert(b);
		g[b].insert(a);
	}
	bfs(k);
	return 0;
}