kuangbin专题一简单搜索 k 迷宫问题

K - 迷宫问题

定义一个二维数组：

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)

题解：

不得不说这是一个好题。
bfs求最短路，dfs求最短路路径。
bfs搜索前最好把路径初始化为INF.
学到了很多。
vector保存最短路路径。

代码：

#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>
#include <queue>
#include <cstdio>
using namespace std;

typedef pair<int,int> P;
const int INF = 0x3f3f3f3f;
int maze[5][5];
int d[5][5];
int dx[4]={1,0,-1,0},dy[4]={0,1,0,-1};
vector<P> ans;
int bfs()
{
    memset(d,0x3f,sizeof(d));
    queue<P> que;
    que.push(P(0,0));
    d[0][0]=0;

    while(que.size())
    {
        P p = que.front();que.pop();

        if(p.first==4&&p.second==4) break;

        for(int i=0;i<4;i++)
        {
            int nx = p.first+dx[i],ny = p.second+dy[i];

            if(0<=nx&&nx<5&&0<=ny&&ny<5&&maze[nx][ny]==0&&d[nx][ny]==INF)
            {
                que.push(P(nx,ny));
                d[nx][ny]=d[p.first][p.second]+1;
            }
        }
    }
    return d[4][4];
}

void dfs_Path(int x,int y)
{
    ans.push_back(P(x,y));
    for(int i=0;i<4;i++)
    {
       int nx = x+dx[i],ny = y+dy[i];
       //printf("%d %d\n",nx,ny);
       //printf("%d %d\n",d[nx][ny],d[x][y]);
       if(nx>=0&&nx<5&&ny>=0&&ny<5&&(d[nx][ny]==d[x][y]-1))
       {
           dfs_Path(nx,ny);
       }
    }
}

int main()
{
    for(int i=0;i<5;i++)
        for(int j=0;j<5;j++)
    {
        scanf("%d",&maze[i][j]);
    }
    bfs();
    dfs_Path(4,4);
    int siz = ans.size();
    for(int i=siz-1;i>=0;i--)
    {
      printf("(%d, %d)\n",ans[i].first,ans[i].second);
    }

    return 0;
}