魔法之森的蘑菇 三维bfs 最短路

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

const int N = 1010;

char g[N][N][4]; // 存储森林地图，g[i][j][p]表示位置(i, j)在面向方向p时的字符值

int px[4] = {-1,0,1,0}; // 上下左右四个方向的x坐标变化量
int py[4] = {0,1,0,-1}; // 上下左右四个方向的y坐标变化量

/*
 *
 * dp[i][j][k]表示在网格中位置为(i, j)且面向方向k时的最小步数
 * 这里的k可以是0、1、2或3，分别代表上、右、下、左四个方向。
 *
 * 面向方向k指的是在迷宫或网格中，当前位置的字符所朝向的方向
 * 如果字符面向上，那么它的下一步移动将是基于向上的方向进行的
 */

int dp[N][N][4]; // dp[i][j][p]表示位置(i, j)在面向方向p时的最短步数

void solve()
{
    int n,m; cin >> n >> m; // 输入矩阵的行数和列数
    int sx,sy,ex,ey; // 分别表示起点和终点的坐标

    /*
    * 通过将 dp 的所有元素初始化为一个较大的值
    * 可以确保在后续的计算过程中
    * 任何实际计算出的步数都会比这个初始值小
    * 从而能够正确地更新 dp 数组中的最小步数值。
    */

    memset(dp,0x3f,sizeof dp); // 初始化dp数组为一个较大的值
    for(int i = 1 ; i <= n ; i ++)
        for(int j = 1 ; j <= m ; j ++)
        {
            char c; cin >> c; // 输入当前位置的字符值

            /*
             * g[i][j][p] 表示位置为 (i, j) 且面向方向为 p 时的字符值
             * 这里的 p 可以是 0、1、2 或 3，分别代表上、右、下、左四个方向。
             */

            for(int p = 0 ; p < 4 ; p ++)
                g[i][j][p] = c; // 存储当前位置在各个方向上的字符值
            if(g[i][j][1] == 'S') // 如果当前位置是起点，记录起点坐标
                sx = i,sy = j;
            if(g[i][j][1] == 'T') // 如果当前位置是终点，记录终点坐标
                ex = i,ey = j;
        }
    for(int i = 0 ; i < 4 ; i ++)
        dp[sx][sy][i] = 0; // 将起点的最短步数初始化为0

    queue<vector<int>> q; // 使用队列进行广度优先搜索
    for(int i = 0 ; i < 4 ; i ++)
        q.push({sx,sy,i}); // 将起点加入队列
    while(q.size()) // 当队列不为空时，继续搜索
    {
        auto t = q.front(); // 取出队首元素
        q.pop(); // 弹出队首元素
        int x = t[0],y = t[1],p = t[2]; // 分别表示当前位置的坐标和面向方向
        int xx = x + px[p] ,yy = y + py[p]; // 计算下一个位置的坐标
        if(xx <= 0 || xx > n || yy <= 0 || yy > m || g[xx][yy][p] =='#') // 如果下一个位置越界或者遇到障碍物，跳过
            continue;
        if(g[xx][yy][p] == '.') // 如果下一个位置是道路，更新最短步数并加入队列
        {
            if(dp[xx][yy][p] > dp[x][y][p] + 1)
            {
                dp[xx][yy][p] = dp[x][y][p] + 1;
                q.push({xx,yy,p});
            }
        }
        else // 如果下一个位置是蘑菇，尝试改变方向并更新最短步数，然后加入队列
        {
            for(int i = 0 ; i < 4 ; i ++)
            {
                if(dp[xx][yy][i] > dp[x][y][p] + 1)
                {
                    dp[xx][yy][i] = dp[x][y][p] + 1;
                    q.push({xx,yy,i});
                }
            }
        }
    }
    int res = 0x3f3f3f3f; // 初始化结果为一个较大的值
    for(int i = 0 ; i < 4 ; i ++) // 遍历四个方向，找到最小的最短步数作为结果
        res = min(dp[ex][ey][i],res);
    if(res != 0x3f3f3f3f) cout << res << '\n'; // 如果结果不是初始值，输出结果
    else cout << -1 << '\n'; // 如果结果是初始值，说明无法到达终点，输出-1
}

int main()
{
    int t; cin >> t; // 输入测试数据组数
    while( t --) 
    {
        solve(); 
    }
}