POJ 2251 Dungeon Master（BFS）

POJ 2251 Dungeon Master
三维迷宫的bfs，设置三维迷宫数组、移动方向数组、访问标记数组，然后bfs即可

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

int vis[35][35][35];
char maze[35][35][35];
struct node{
    int x,y,z,step;
};
queue<node> q;

int l,r,c,sx,sy,sz,gx,gy,gz;
int dir[6][3]={{0,0,-1},{0,0,1},{0,-1,0},{0,1,0},{1,0,0},{-1,0,0}};

//移动合法性检查 
int check(int x,int y,int z){
    if(x<0||y<0||z<0||x>=l||y>=r||z>=c||maze[x][y][z]=='#'||vis[x][y][z]==1)
        return 0;
    return 1;
}

int bfs(){
    while(!q.empty())q.pop();
    node start,next;
    start.x=sx;
    start.y=sy;
    start.z=sz;
    start.step=0;
    vis[sx][sy][sz]=1;
    q.push(start);
    while(!q.empty()){
        start=q.front();
        q.pop();
        //终止判断 
        if(start.x==gx&&start.y==gy&&start.z==gz)
            return start.step;
        for(int i=0;i<6;i++){
            next=start;
            next.x=start.x+dir[i][0];
            next.y=start.y+dir[i][1];
            next.z=start.z+dir[i][2];
            if(check(next.x,next.y,next.z)){
                vis[next.x][next.y][next.z]=1;
                next.step=start.step+1;
                q.push(next);
            }
        }
    }
    return 0;
}

int main(){
//  freopen("input_B_Dungeon.txt","r",stdin);
    while(scanf("%d%d%d",&l,&r,&c)){
        if(l==0&&r==0&&c==0)break;
        for(int i=0;i<l;i++)
            for(int j=0;j<r;j++){
                scanf("%s",maze[i][j]);             //scanf遇到空格结束输入 
                for(int k=0;k<c;k++){
                    if(maze[i][j][k]=='S'){
                        sx=i;
                        sy=j;
                        sz=k;
                    }
                    if(maze[i][j][k]=='E'){
                        gx=i;
                        gy=j;
                        gz=k;
                    }
                }
            }
        memset(vis,0,sizeof(vis));
        int sum=bfs();
        if(sum){
            printf("Escaped in %d minute(s).\n",sum);
        }
        else{
            printf("Trapped!\n");
        }                   
    }   
    return 0;
}

为有牺牲多壮志，敢叫日月换新天！