【kuangbin】简单搜索 - 2.地牢大师【BFS】

Date：2022.01.31
题意：
你现在被困在一个三维地牢中，需要找到最快脱离的出路！
地牢由若干个单位立方体组成，其中部分不含岩石障碍可以直接通过，部分包含岩石障碍无法通过。
向北，向南，向东，向西，向上或向下移动一个单元距离均需要一分钟。
你不能沿对角线移动，迷宫边界都是坚硬的岩石，你不能走出边界范围。
请问，你有可能逃脱吗？
如果可以，需要多长时间？
输入格式
输入包含多组测试数据。
每组数据第一行包含三个整数 L,R,C 分别表示地牢层数，以及每一层地牢的行数和列数。
接下来是 L 个 R 行 C 列的字符矩阵，用来表示每一层地牢的具体状况。
每个字符用来描述一个地牢单元的具体状况。
其中, 充满岩石障碍的单元格用”#”表示，不含障碍的空单元格用”.”表示，你的起始位置用”S”表示，终点用”E”表示。
每一个字符矩阵后面都会包含一个空行。
当输入一行为”0 0 0”时，表示输入终止。
输出格式
每组数据输出一个结果，每个结果占一行。
如果能够逃脱地牢，则输出”Escaped in x minute(s).”，其中X为逃脱所需最短时间。
如果不能逃脱地牢，则输出”Trapped!”。
数据范围
≤L,R,C≤100
输入样例：
3 4 5
S…
.###.
.##…
###.#

##.##
##…

#.###
####E

1 3 3
S##
#E#

0 0 0
输出样例：
Escaped in 11 minute(s).
Trapped!

思路：三维方向BFS，每个格子看成正方体，最多有六个方向可走。
代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 150;
char s[N][N][N];
LL l,r,c,stx,sty,stz,edx,edy,edz,d[N][N][N];
LL dx[6]={0,0,-1,1,0,0},dy[6]={1,-1,0,0,0,0},dz[6]={0,0,0,0,1,-1};
struct node
{
    LL z,x,y;
};
void bfs(LL stz,LL stx,LL sty)
{
    for(int i=0;i<l;i++)
        for(int j=0;j<r;j++)
            for(int k=0;k<c;k++) d[i][j][k]=-1;
    queue<node>q;d[stz][stx][sty]=0;q.push({stz,stx,sty});
    while(q.size())
    {
        node t=q.front();q.pop();
        for(int i=0;i<6;i++)
        {
            LL iz=t.z+dz[i],iy=t.y+dy[i],ix=t.x+dx[i];
            if(ix>=r||ix<0||iy<0||iy>=c||iz<0||iz>=l||d[iz][ix][iy]!=-1||s[iz][ix][iy]=='#') continue;
            d[iz][ix][iy]=d[t.z][t.x][t.y]+1;
            q.push({iz,ix,iy});
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    while(cin>>l>>r>>c)
    {
        if(l==0) break;
        for(int i=0;i<l;i++)
            for(int j=0;j<r;j++)
                for(int k=0;k<c;k++) 
                {
                    cin>>s[i][j][k];
                    if(s[i][j][k]=='S') {stz=i;stx=j;sty=k;}
                    else if(s[i][j][k]=='E') {edz=i;edx=j;edy=k;}
                }
        bfs(stz,stx,sty);
        if(d[edz][edx][edy]==-1) cout<<"Trapped!\n";
        else cout<<"Escaped in "<<d[edz][edx][edy]<<" minute(s).\n";
    }
    return 0;
}