题目描述:
给定一个 n×n 的二维数组,如下所示:
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表示可以走的路,只能横着走或竖着走,不能斜着走,要求编程序找出从左上角到右下角的最短路线。
数据保证至少存在一条从左上角走到右下角的路径。
输入格式
第一行包含整数 n。
接下来 n 行,每行包含 n 个整数 0 或 1,表示迷宫。
输出格式
输出从左上角到右下角的最短路线,如果答案不唯一,输出任意一条路径均可。
按顺序,每行输出一个路径中经过的单元格的坐标,左上角坐标为 (0,0),右下角坐标为 (n−1,n−1)。
数据范围
0≤n≤1000
输入样例:
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
输出样例:
0 0
1 0
2 0
2 1
2 2
2 3
2 4
3 4
4 4
分析:
本题是BFS求最短路径的基本问题,不过输出的是最短路径而不是路径的长度,所以可以在bfs过程中记录下当前坐标的上一步坐标,为了方便输出,可以从终点开始bfs到起点倒着遍历。同时可以用pre数组作为当前坐标是否已被访问的标志数组,当遍历到起点时,可以直接return来剪枝。
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1005;
typedef pair<int,int> PII;
int g[N][N],n;
PII q[N*N],pre[N][N];
int dx[] = {0,0,1,-1};
int dy[] = {1,-1,0,0};
void bfs(int x,int y){
int hh = 0,tt = 0;
q[0] = {x,y};
memset(pre,-1,sizeof pre);
pre[x][y] = {0,0};
while(hh <= tt){
PII t = q[hh++];
for(int i = 0;i < 4;i++){
int nx = t.first + dx[i],ny = t.second + dy[i];
if(g[nx][ny] || pre[nx][ny].first != -1 || nx < 0 || nx >= n || ny < 0 || ny >=n) continue;
q[++tt] = {nx,ny};
pre[nx][ny] = t;
if(!nx && !ny) return;
}
}
}
int main(){
scanf("%d",&n);
for(int i = 0;i < n;i++){
for(int j = 0;j < n;j++){
scanf("%d",&g[i][j]);
}
}
bfs(n - 1,n - 1);
PII t(0,0);
while(1){
printf("%d %d\n",t.first,t.second);
if(t.first == n - 1 && t.second == n - 1) break;
t = pre[t.first][t.second];
}
return 0;
}