水管工游戏——DFS实现

题目描述

一块矩形土地被分为NM的单位正方形，现在这块土地上已经埋设有一些水管，水管将从坐标为(1,1)左上角左部边缘，延伸到(N,M)右下角右部边缘。水管只有2种，如下图所示。

每种管道将占据一个单位正方形土地。你现在可以旋转这些管道，使得构成一个管道系统，即创造一条从(1,1)到(N,M)的连通管道。标有树木的方格表示这里没有管道。如下图：一个54的土地中(2,4)处有一个树木。

我们可以旋转其中的一些管道，使之构成一个连通的管道系统，如下图。

如果通过旋转管道可以使之构成一个连通的管道系统，就输出铺设的路径，
否则输出impossible。

输入：第一行为两个整数N和M（都不超过10），接下来的N行，每行都有M个整数，表示地图中的每一小格，其中0表示树木，1~6分别表示管道的六种不同的摆放方式。

输入样例

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

输出样例

（1,1）（1,2）（2,2）（3,2）（3,3）（3,4）（4,4）（5,4）

#include<bits/stdc++.h>
using namespace std;
int n,m;
int a[100][100];
int vist[100][100];
int top=0;
int flag;
struct node{
	int x;
	int y;
};
node s[100];
void dfs(int x,int y,int direction){
	if(x==n&&y==m+1){
		flag=1;
		for(int i=1;i<=top;i++){
			cout<<"("<<s[i].x<<","<<s[i].y<<")"<<" ";
		}
		return;
	}
	if(x<1||x>n||y<1||y>n)
	return;
	if(vist[x][y]){
		return;
	}
	vist[x][y]=1;
	top++;
	s[top].x=x;
	s[top].y=y;
	//如果是直管
	if(a[x][y]==5||a[x][y]==6){
		if(direction==1){//进水口在左边 
			dfs(x,y+1,1);
		}
		if(direction==2){//进水口在上边 
			dfs(x+1,y,2);
		}
		if(direction==3){//进水口在右边 
			dfs(x,y-1,3);
		}
		if(direction==4){//进水口在下边 
			dfs(x+1,y,4);
		}
	} 
	if(a[x][y]>=1&&a[x][y]<=4){
		if(direction==1){//进水口在左边 
			dfs(x+1,y,2);
			dfs(x-1,y,4);
		}
		if(direction==2){//进水口在上边 
			dfs(x,y+1,1);
			dfs(x,y-1,3);
		}
		if(direction==3){//进水口在右边 
			dfs(x+1,y,2);
			dfs(x-1,y,4);
		}
		if(direction==4){//进水口在下边 
			dfs(x,y+1,1);
			dfs(x,y-1,3);
		}
	}
	vist[x][y]=0;
	top--;
	return; 
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	for(int j=1;j<=m;j++){
		cin>>a[i][j];
	}
	dfs(1,1,1);
	if(!flag){
		cout<<"impossible";
	}
}