POJ 2251 Dungeon Master bfs 难度:0

http://poj.org/problem?id=2251

bfs,把两维换成三维,但是30*30*30=9e3的空间时间复杂度仍然足以承受

 

#include <cstdio>

#include <cstring>

#include <algorithm>

#include <queue>

using namespace std;

struct P{

        int x,y,z;

        P(){x=y=z=0;}

        P(int x,int y,int z){

                this->x=x;this->y=y;this->z=z;

        }

};

const int dx[6]={1,-1,0,0,0,0};

const int dy[6]={0,0,1,-1,0,0};

const int dz[6]={0,0,0,0,1,-1};



int l,r,c;

char maz[30][30][31];

int step[30][30][30];

bool in(int z,int x,int y){

        return z>=0&&z<l&&

        x>=0&&x<r&&

        y>=0&&y<c;

}

int main(){

        while(scanf("%d%d%d",&l,&r,&c)==3&&l){

                memset(step,0x7f,sizeof(step));

                for(int i=0;i<l;i++){

                        for(int j=0;j<r;j++){

                                scanf("%s",maz[i][j]);

                        }

                }

                queue<P> que;

                for(int i=0;i<l;i++){

                        for(int j=0;j<r;j++){

                                for(int k=0;k<c;k++){

                                        if(maz[i][j][k]=='E'){

                                                que.push(P(j,k,i));

                                                step[i][j][k]=0;

                                        }

                                }

                        }

                }

                bool fl=false;

                while(!que.empty()){

                        P f=que.front();que.pop();

                        int z=f.z,x=f.x,y=f.y;

                        if(maz[z][x][y]=='S'){

                                fl=true;

                                printf("Escaped in %d minute(s).\n",step[z][x][y]);

                                break;

                        }

                        for(int i=0;i<6;i++){

                                int tx=x+dx[i],ty=y+dy[i],tz=z+dz[i];

                                if(in(tz,tx,ty)&&maz[tz][tx][ty]!='#'&&step[tz][tx][ty]>step[z][x][y]+1){

                                        step[tz][tx][ty]=step[z][x][y]+1;

                                        que.push(P(tx,ty,tz));

                                }

                        }

                }

                while(!que.empty())que.pop();

                if(!fl)puts("Trapped!");

        }

        return 0;

}