国王的城堡(bfs专题)

题面

warma被魔王抓走了,有一天魔王出差去了,warma觉得这是逃亡的好机会.

魔王的城堡是一个ABC的立方体,可以被表示成A个B*C的矩阵,刚开始warma被关在(0,0,0)的位置,离开城堡的门在(A-1,B-1,C-1)的位置,现在知道魔王将在T分钟后回到城堡,warma每分钟能从一个坐标走到相邻的六个坐标中的其中一个.现在给你城堡的地图,请你计算出warma能否在魔王回来前离开城堡(只要走到出口就算离开城堡,如果走到出口的时候魔王刚好回来也算逃亡成功),如果可以请输出需要多少分钟才能离开,如果不能则输出-1.

输入：

输入数据的第一行是一个正整数K,表明测试数据的数量.每组测试数据的第一行是四个正整数A,B,C和T(1<=A,B,C<=50,1<=T<=1000),它们分别代表城堡的大小和魔王回来的时间.然后是A块输入数据(先是第0块,然后是第1块,第2块…),每块输入数据有B行,每行有C个正整数,代表迷宫的布局,其中0代表路,1代表墙.(如果对输入描述不清楚,可以参考Sample Input中的迷宫描述,它表示的就是上图中的迷宫) 注意:起点本身也可能是一堵墙即（0,0,0）位置上的数可能是1。

输出：

对于每组测试数据,如果Ignatius能够在魔王回来前离开城堡,那么请输出他最少需要多少分钟,否则输出-1.

样例输入：
1
3 3 4 20
0 1 1 1
0 0 1 1
0 1 1 1
1 1 1 1
1 0 0 1
0 1 1 1
0 0 0 0
0 1 1 0
0 1 1 0
样例输出：
11

解题思路

其实就是简单的BFS，只是在二维迷宫的基础上拓展为三维。

代码

#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;

int mp[55][55][55];
int bj[55][55][55];
int mov[6][3] = {1,0,0,-1,0,0,0,1,0,0,-1,0,0,0,1,0,0,-1};
struct Node
{
    int k,x, y, step;
} a,b ;
queue<Node> ppp;
int main ()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int flag=0;
        memset(bj, 0, sizeof(bj));
        int h,n,m,minn=10000,shijian;
        scanf("%d %d %d %d",&h,&n,&m,&shijian);
        minn=10000;
        for(int k=0; k<h; k++)
            for(int i=0; i<n; i++)
                for(int j=0; j<m; j++)
                    scanf("%d",&mp[k][i][j]);
        a.x=0;
        a.y=0;
        a.k=0;
        a.step=0;
        ppp.push(a);
        bj[a.x][a.y][a.k]=1;
        while(!ppp.empty())
        {
            a=ppp.front();
            ppp.pop();
            if(a.k==h-1&&a.x==n-1&&a.y==m-1&&a.step<=shijian)
            {
                printf("%d\n",a.step);
                flag=1;
                break;
            }
            b.step=a.step+1;
            for(int i=0; i<6; i++)
            {
                b.k=a.k+mov[i][0];
                b.x=a.x+mov[i][1];
                b.y=a.y+mov[i][2];
                if(b.k<0||b.k>=h||b.x<0||b.x>=n||b.y<0||b.y>=m||bj[b.k][b.x][b.y]||mp[b.k][b.x][b.y])
                    continue;
                bj[b.k][b.x][b.y]=1;
                ppp.push(b);
            }
        }
        if(flag==0)
            printf("-1\n");
    }
    return 0;
}