POJ2251——Dungeon Master（三维BFS）

Posted 2020-09-20 Ying_zx

和迷宫问题区别不大，相比于POJ1321的棋盘问题，这里的BFS是三维的，即从4个方向变为6个方向。

用上队列的进出操作较为轻松。

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

char map[35][35][35];
int vis[35][35][35];
int to[6][3]={{0,0,1},{0,0,-1},{0,1,0},{0,-1,0},{1,0,0},{-1,0,0}};
int k,n,m,sx,sy,sz,ex,ey,ez;
struct node
{
    int x,y,z,step;
};
int check(int x,int y,int z)
{
    if(x<0||y<0||z<0||x>=k||y>=n||z>=m||map[x][y][z]==‘#‘||vis[x][y][z])
        return 1;
    return 0;
}
int bfs()
{
    int i;
    node a,next;
    queue<node> Q;
    a.x = sx,a.y= sy,a.z = sz;
    a.step = 0;
    vis[sx][sy][sz]=1;
    Q.push(a);
    while(!Q.empty())
    {
        a=Q.front();
        Q.pop();
        if(a.x==ex&&a.y==ey&&a.z==ez)
            return a.step;
        for(i=0;i<6;i++)
        {
            next=a;
            next.x=a.x+to[i][0];
            next.y=a.y+to[i][1];
            next.z=a.z+to[i][2];
            if(check(next.x,next.y,next.z))
                continue;
            vis[next.x][next.y][next.z]=1;
            next.step=a.step+1;
            Q.push(next);
        }
    }
    return 0;
}
int main()
{
    int i,j,r;
    while(scanf("%d%d%d",&k,&n,&m),n+m+k)
    {
        for(i=0;i<k;i++)
        {
            for(j=0;j<n;j++)
            {
                scanf("%s",map[i][j]);
                for(r=0;r<m;r++)
                {
                    if(map[i][j][r]==‘S‘)
                    {
                        sx=i,sy=j,sz=r;
                    }
                    else if(map[i][j][r]==‘E‘)
                    {
                        ex=i,ey=j,ez=r;
                    }
                }
            }
        }
        memset(vis,0,sizeof(vis));
        int ans;
        ans=bfs();
        if(ans)  
            printf("Escaped in %d minute(s).\n",ans);  
        else  
            printf("Trapped!\n");  
    }
    return 0;
}