poj 2251 Dungeon Master(广搜)

题意:三维空间,可以走上下左右前后六个方向,求最短路径,BFS

#include<stdio.h>
#include<queue>
#include<string.h>
using namespace std;
const int MAXN=50;

char str[MAXN][MAXN][MAXN];
int step[MAXN][MAXN][MAXN];
int vis[MAXN][MAXN][MAXN];
int l,r,c;
struct Node
{
    int x,y,z;
}s,e;

int BFS()
{
    int i,j;
    Node head,next;
    queue<Node>Q;
    Q.push(s);
    step[s.x][s.y][s.z]=0;
    while(!Q.empty())
    {
        head=Q.front();
        Q.pop();
        for(i=0;i<6;i++)
        {
            next=head;
            if(i==0) next.x-=1;
            if(i==1) next.x+=1;
            if(i==2) next.y-=1;
            if(i==3) next.y+=1;
            if(i==4) next.z-=1;
            if(i==5) next.z+=1;
            if(!vis[next.x][next.y][next.z] && str[next.x][next.y][next.z]==1)
            {
                vis[next.x][next.y][next.z]=1;
                Q.push(next);
                step[next.x][next.y][next.z]=step[head.x][head.y][head.z]+1;
                if(next.x==e.x && next.y==e.y && next.z==e.z) return  step[head.x][head.y][head.z]+1;
            }
        }
    }
    return  0;
}
int main()
{
    int i,j,k;
    while(scanf("%d%d%d",&l,&r,&c)!=EOF)
    {
        if(l==0 && r==0 && c==0) break;
        memset(vis,0,sizeof(vis));
        memset(str,0,sizeof(str));
        for(i=1; i<=l; i++)
        {
            for(j=1; j<=r; j++)
            {
                scanf("%s",str[i][j]+1);
                for(k=1; str[i][j][k]; k++)
                {
                    if(str[i][j][k]=='#') str[i][j][k]=0;
                    if(str[i][j][k]=='.') str[i][j][k]=1;
                    if(str[i][j][k]=='S')
                    {
                        str[i][j][k]=1;
                        s.x=i;
                        s.y=j;
                        s.z=k;
                    }
                    if(str[i][j][k]=='E')
                    {
                        str[i][j][k]=1;
                        e.x=i;
                        e.y=j;
                        e.z=k;
                    }
                }
            }
        }
        int ans=BFS();
        if(ans) printf("Escaped in %d minute(s).\n",ans);
        else printf("Trapped!\n");

    }
    return 0;
}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券