PTA 7-5 地下迷宫探索 (30分)
地道战是在抗日战争时期,在华北平原上抗日军民利用地道打击日本侵略者的作战方式。地道网是房连房、街连街、村连村的地下工事,如下图所示。
我们在回顾前辈们艰苦卓绝的战争生活的同时,真心钦佩他们的聪明才智。在现在和平发展的年代,对多数人来说,探索地下通道或许只是一种娱乐或者益智的游戏。本实验案例以探索地下通道迷宫作为内容。
假设有一个地下通道迷宫,它的通道都是直的,而通道所有交叉点(包括通道的端点)上都有一盏灯和一个开关。请问你如何从某个起点开始在迷宫中点亮所有的灯并回到起点?
样例图的测试数据:
8 10 1
1 2
2 6
6 4
4 7
7 8
3 4
4 5
1 5
3 5
1 7
深度遍历的变形,只要每条分支遍历到尽头时将其节点在入栈一次即可
#include<iostream>
#include<vector>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<set>
using namespace std;
int vis[1001]={0};
set<int> s;
vector<int> v[1001],v2;
void dfs(int S)
{
v2.push_back(S);
vis[S]=1;
for(int j=0;j<v[S].size();j++)
{
if(vis[v[S][j]]==0)
{
vis[v[S][j]]=1;
dfs(v[S][j]);
v2.push_back(S);//push进上一层节点,保证没有重复的节点
}
}
}
int main()
{
int N,M,S;
cin>>N>>M>>S;
for(int i=0;i<M;i++)
{
int v1,v2;
cin>>v1>>v2;
v[v1].push_back(v2);
v[v2].push_back(v1);
}
for(int i=1;i<=N;i++)
{
sort(v[i].begin(),v[i].end());
}
fill(vis+1,vis+N+1,0);
dfs(S);
for(int i=0;i<v2.size()-1;i++)
{
printf("%d ",v2[i]);
}
printf("%d",v2[v2.size()-1]);
for(int i=1;i<=N;i++)
{
if(vis[i]==0)//有未标记的说明不连通
{
printf(" %d",0);
return 0;
}
}
return 0;
}