5-33 地下迷宫探索 (30分) -- dfs

5-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

 

• 时间限制：400ms
• 内存限制：64MB
• 代码长度限制：16kB
• 判题程序：系统默认
• 作者：DS课程组
• 单位：浙江大学

http://pta.patest.cn/pta/test/15/exam/4/question/860

#include <cstdio>  
#include <sstream>  
#include <cstring>  
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <queue>

using namespace std;

#define  N 1005

int n , m , s ;

vector<int> vt[N] ;

bool cmp(int x ,int y)
{
	return x < y ;
}

bool vis[N] ;
deque<int> que ;
bool flag ; // 输出控制
int outLen ; // 判断是否可以遍历所有节点

void dfs()
{
	int sta = que.back() ;
	if(sta == s && vis[s])
		return ;
	int len = vt[sta].size();
	if(len > 1)
		sort(vt[sta].begin() ,vt[sta].end() , cmp);

	for(int i = 0 ; i < len ;i++)
	{
		int nextNum = vt[sta][i] ;
		if(!vis[nextNum])
		{
			if(!flag) // 打印入队列的路径
			{
				printf("%d",sta) ;
				flag = true ;
				outLen ++ ;
			}else{
				printf(" %d",sta) ;
				outLen ++ ;
			}
			que.push_back(nextNum) ;
			vis[sta] = true ;
			dfs() ;
		}
	}
	// 打印回去的路径
	if(!que.empty()) // 当dfs 找不到的时候 就出队列 并设置为true 输出
	{
		int top = que.back() ;
		que.pop_back() ;
		vis[top] = true ;
		if(!flag)
		{
			printf("%d",top) ;
			flag = true ;
			outLen ++ ;
		}else{
			printf(" %d",top) ;
			outLen ++ ;
		}
	}
}

int main()
{
    //freopen("in.txt", "r", stdin);
	while( scanf("%d%d%d", &n,&m,&s) != EOF)
	{
		int i ;
		int v1 , v2 ;
		for(i = 1 ; i <= n ; i++)
		{
			vt[i].clear() ;
		}
		for(i = 0 ;i < m ; i++)
		{
			scanf("%d%d" ,  &v1 , &v2);
			vt[v1].push_back(v2) ;
			vt[v2].push_back(v1) ;
		}

		for(i = 1 ; i <= n ; i++)
		{
			vis[i] = false ;
		}
		while(!que.empty())
		{
			que.pop_back();
		}
		que.push_back(s) ;
		flag = false ;
		outLen = 0 ;
		dfs();
		if(outLen != 2*n -1)
			printf(" 0") ;
		printf("\n") ;
	}
	return 0 ;
}