POJ 3984 迷宫问题

迷宫问题

Time Limit: 1000MS		Memory Limit: 65536K
Total Submissions: 49149		Accepted: 26684

Description

定义一个二维数组： 

int maze[5][5] = {

	0, 1, 0, 0, 0,

	0, 1, 0, 1, 0,

	0, 0, 0, 0, 0,

	0, 1, 1, 1, 0,

	0, 0, 0, 1, 0,

};

它表示一个迷宫，其中的1表示墙壁，0表示可以走的路，只能横着走或竖着走，不能斜着走，要求编程序找出从左上角到右下角的最短路线。

Input

一个5 × 5的二维数组，表示一个迷宫。数据保证有唯一解。

Output

左上角到右下角的最短路径，格式如样例所示。

Sample Input

0 1 0 0 0
0 1 0 1 0
0 0 0 0 0
0 1 1 1 0
0 0 0 1 0

Sample Output

(0, 0)
(1, 0)
(2, 0)
(2, 1)
(2, 2)
(2, 3)
(2, 4)
(3, 4)
(4, 4)

#include <iostream>
#include <cstdio>
#include <queue>

using namespace std;

typedef pair<int,int> P;

const int INF = 1e8;

const int N=5,M=5;
int maze[N][M],d[N][M];
int sx=0,sy=0;
int gx=4,gy=4;

int dx[2]={0,1},dy[2]={1,0};

struct
{
    int x;
    int y;
}pre[N][M];

void bfs()
{
    // 初始化队列
    queue<P> que;
    // 初始化距离
    for(int i=0;i<N;++i)
    {
        for(int j=0;j<M;++j)
        {
            d[i][j]=INF;
        }
    }
    d[0][0]=0;
    que.push(P(0,0));
    //
    while(que.size())
    {
        // 取出元素
        P p=que.front();
        que.pop();
        // 找到终点
        if(p.first==gx && p.second==gy)
        {
            break;
        }
        // 左和下找其余可行顶点
        for(int i=0;i<2;++i)
        {

            int nx=p.first+dx[i];
            int ny=p.second+dy[i];
            //printf("%d %d
",nx,ny);
            if(nx>=0 && nx<N && ny>=0 && ny<M && maze[nx][ny]==0 && d[nx][ny]==INF)
            {
                que.push(P(nx,ny));
                d[nx][ny]=d[p.first][p.second]+1;
                // 记录前一节点
                pre[nx][ny].x=p.first;
                pre[nx][ny].y=p.second;
            }
        }
    }

}

void dfs(int nx,int ny)
{
    if(pre[nx][ny].x==0 && pre[nx][ny].y==0)
    {
        printf("(0, 0)
");
        printf("(%d, %d)
",nx,ny);
        return;
    }

    dfs(pre[nx][ny].x,pre[nx][ny].y);
    printf("(%d, %d)
",nx,ny);

}

void solve()
{
    bfs();
    if(d[gx][gy]==INF)
    {
        printf("NO
");
    }
    else
    {
        dfs(gx,gy);
    }
}

int main()
{
    for(int i=0;i<N;++i)
    {
        for(int j=0;j<M;++j)
        {
            scanf("%d",&maze[i][j]);
        }
    }
    solve();
    return 0;
}