PTA 数据结构与算法编程集 7-33 地下迷宫探索

7-33 地下迷宫探索 (30 分)地道战是在抗日战争时期,在华北平原上抗日军民利用地道打击日本侵略者的作战方式。地道网是房连房、街连街、村连村的地下工事,如下图所示。

我们在回顾前辈们艰苦卓绝的战争生活的同时,真心钦佩他们的聪明才智。在现在和平发展的年代,对多数人来说,探索地下通道或许只是一种娱乐或者益智的游戏。本实验案例以探索地下通道迷宫作为内容。
假设有一个地下通道迷宫,它的通道都是直的,而通道所有交叉点(包括通道的端点)上都有一盏灯和一个开关。请问你如何从某个起点开始在迷宫中点亮所有的灯并回到起点?

输入格式:
输入第一行给出三个正整数,分别表示地下迷宫的节点数N(1<N≤1000,表示通道所有交叉点和端点)、边数M(≤3000,表示通道数)和探索起始节点编号S(节点从1到N编号)。随后的M行对应M条边(通道),每行给出一对正整数,分别是该条边直接连通的两个节点的编号。
输出格式:
若可以点亮所有节点的灯,则输出从S开始并以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 3 4 5 6 5 4 3 2 1

输入样例2:
6 6 6
1 2
1 3
2 3
5 4
6 5
6 4
输出样例2:
6 4 5 4 6 0

一开始用的dfs进行遍历,写完之后只能过两个测试点,后两个测试点没法通过,我左想右想觉得我没什么问题,想了很久没想出来只好去看别人得代码,看完之后恍然大悟,知道自己错哪了
下面是一开始错误的代码:

#include<iostream>
#include<cstring>
#include<stack> 
using namespace std;
int N,M,S;
int vertex[1010],path[1010][1010],vis[1010];
stack<int> p;
void dfs(int st)
{
 vis[st] = 1;
 for(int i=1;i <= N;i++){
  if(!vis[i] && path[st][i] != 0){
   printf(" %d",i);
   p.push(i);
   dfs(i);
  }
 }
}
int main()
{
 int number;
 scanf("%d%d%d",&N,&M,&S);
 for(int i=0;i < M;i++){
  int u,v;
  scanf("%d%d",&u,&v);
  path[u][v] = path[v][u] = 1;
 }
 p.push(S);
 cout << S;
 dfs(S);
 p.pop();   //栈中的元素 等于N-1 
 number = p.size();
 while(!p.empty()){ //按原路返回
  printf(" %d",p.top());
  p.pop();
 }
 if(number < N-1)
  cout << " 0";
 return 0;
}

按图的遍历的话是没问题的,但是这是从实际问题出发的dfs,也就是在第一次深度优先遍历结束后人是要按原路返回到上一个结点进行下一次深度优先遍历,而没办法直接跳到上一个结点
以下是参照别人的想法而改进的代码:

#include<iostream>
#include<cstring>
using namespace std;
int N,M,S;
int path[1010][1010],vis[1010],tag=0,cnt = 1;
void dfs(int st)
{
 if(tag++)
  cout << " ";
 printf("%d",st);
 for(int i=1;i <= N;i++){
  if(!vis[i] && path[st][i] != 0){
   vis[i] = 1;
   cnt++;
   dfs(i);
   printf(" %d",st);
  }
 }
}
int main()
{
 scanf("%d%d%d",&N,&M,&S);
 for(int i=0;i < M;i++){
  int u,v;
  scanf("%d%d",&u,&v);
  path[u][v] = path[v][u] = 1;
 }
 vis[S] = 1;
 dfs(S);
 if(cnt < N)
  printf(" 0");
 return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_43189757/article/details/88685865