定义一个二维数组:
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,沿途记录各个结点的父节点信息,最后打印即可
#include<cstdio>
#include<string.h>
#include<queue>
#include<vector>
#include<map>
#include<stack>
#include<set>
#include<algorithm>
#include<map>
#include<math.h>
using namespace std;
#define INF 1<<28
static const int MAX = 1050;
int dx[4] = {-1, 1, 0, 0};
int dy[4] = {0, 0, -1, 1};
int maze[10][10];
int visit[10][10];
int px[10][10];
int py[10][10];
int step[10][10];
void bfs()
{
memset(visit, 0, sizeof(visit));
memset(px, -1, sizeof(px));
memset(py, -1, sizeof(py));
for (int i = 0; i < 5; i++)
for (int j = 0; j < 5; j++)
step[i][j] = INF;
queue<pair<int, int> > Q;
Q.push(make_pair(0, 0));
step[0][0] = 0;
while (!Q.empty())
{
pair<int, int> q = Q.front(); Q.pop();
visit[q.first][q.second] = 1;
if (q.first == 4 && q.second == 4)
break;
for (int i = 0; i < 4; i++)
{
int tx = q.first + dx[i];
int ty = q.second + dy[i];
if (tx < 0 || tx >= 5 || ty < 0 || ty >= 5)
continue;
if (maze[tx][ty] == 1)
continue;
if (step[tx][ty] > step[q.first][q.second] + 1)
{
step[tx][ty] = step[q.first][q.second] + 1;
px[tx][ty] = q.first;
py[tx][ty] = q.second;
}
if (visit[tx][ty] == 0)
Q.push(make_pair(tx, ty));
}
}
vector<pair<int, int> > path;
path.push_back(make_pair(4, 4));
for (int i = 4, j = 4; px[i][j] != 0 && px[i][j] != 0;)
{
path.push_back(make_pair(px[i][j], py[i][j]));
int u = i;
int v = j;
i = px[u][v];
j = py[u][v];
}
path.push_back(make_pair(0, 0));
reverse(path.begin(), path.end());
for (vector<pair<int, int> >::iterator p = path.begin(); p != path.end(); p++)
{
printf("(%d, %d)\n", (*p).first, (*p).second);
}
}
int main()
{
for (int i = 0; i < 5; i++)
for (int j = 0; j <5; j++)
scanf("%d", &maze[i][j]);
bfs();
return 0;
}