水管工游戏——dfs

• 问题描述：

水管工游戏是指如下图中的矩阵中，一共有两种管道，一个是直的，一个是弯的，所有管道都可以自由旋转，最终就是要连通入水口可出水口。其中的树为障碍物。

方案：

输入格式：输入的第一行为两个整数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)

• 算法思路：

本题使用的是深度优先搜索算法，先标识入水口方向，在每次探索新管道的时候上根据入水口方向进行扩展，然后根据管道的类型来判断下一个节点的入水口方向。水管入口：1代表左，2代表右，3代表上，4代表下。

• 代码：

#include<cstdio>
#include<iostream>
#define INF 10000000
using namespace std;
int n,m,flag=;
int a[][],b[][];
int next[][]={{,},{,-},{,},{-,}};
struct node
{
    int x,y;
}s[];
int top=;
void dfs(int x,int y,int r)
{

    if(x==n&&y==m+)//走到出口
    {
        flag=;
        for(int i=;i<=top;i++)
            printf("(%d,%d)",s[i].x,s[i].y);
        return ;
    }
    if(x<||y<||x>n||y>m)return ;
    if(b[x][y]==)//如果这个点已在路径中
        return ;
    b[x][y]=;//标记
    top++;
    s[top].x=x;
    s[top].y=y;
    //直水管的情况
    if(a[x][y]==||a[x][y]==)
    {
        if(r==)//入口在左
        {
            dfs(x,y+,);
        }
        else if(r==)//入口在右
        {
            dfs(x,y-,);
        }
        else if(r==)//入口在上
        {
            dfs(x+,y,);
        }
        else//入口在下
        {
            dfs(x-,y,);
        }
    }
    //弯水管的情况
    if(a[x][y]>=&&a[x][y]<=)
    {
        if(r==)
        {
            dfs(x+,y,);
            dfs(x-,y,);
        }
        else if(r==)
        {
            dfs(x-,y,);
            dfs(x+,y,);
        }
        else if(r==)
        {
            dfs(x,y+,);
            dfs(x,y-,);
        }
        else
        {
            dfs(x,y+,);
            dfs(x,y-,);
        }
    }
    b[x][y]=;//尝试过不能用的点要取消标记并且出栈
    top--;
    return ;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    dfs(,,);
    if(flag==)
        printf("impossible");
    printf("\n");
    return ;
}