AcWing 1076.迷宫问题 —— 记录路径的bfs

给定一个 n×nn×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)(0,0)，右下角坐标为 (n−1,n−1)。

数据范围

0≤n≤1000

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>

#define x first
#define y second

using namespace std;

typedef pair<int, int> PII;

const int N = 1010;
int g[N][N];
PII pre[N][N];
int n;

void bfs()
{
    int dx[] = {0, 0, -1, 1}, dy[] = {1, -1, 0, 0};
    queue<PII> q;
    q.push({n - 1, n - 1});
    pre[n - 1][n - 1] = {-1, -1};

    while (q.size())
    {
        PII t = q.front();
        q.pop();
        
        for (int i = 0; i < 4; i ++ )
        {
            int x = t.x + dx[i], y = t.y + dy[i];
            if (x < 0 || x >= n || y < 0 || y >= n) continue;
            if (g[x][y] == 1) continue;
            if (pre[x][y].x != 0 || pre[x][y].y != 0) continue;

            pre[x][y] = t;
            if (x == 0 && y == 0) return;
            q.push({x, y});
        }
    }
}

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();

    PII end = {0, 0};

    while (1)
    {
        printf("%d %d\n", end.x, end.y);
        if (end.x == n - 1 && end.y == n - 1) break;
        end = pre[end.x][end.y];
    }

    return 0;
}

记录路径只需使用一个pre数组用来记录当前状态是从那个状态转移过来即可。