胜利大逃亡(HDOJ 1253)

Problem Description

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

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

Input

输入数据的第一行是一个正整数K,表明测试数据的数量.每组测试数据的第一行是四个正整数A,B,C和T(1<=A,B,C<=50,1<=T<=1000),它们分别代表城堡的大小和魔王回来的时间.然后是A块输入数据(先是第0块,然后是第1块,第2块......),每块输入数据有B行,每行有C个正整数,代表迷宫的布局,其中0代表路,1代表墙.(如果对输入描述不清楚,可以参考Sample Input中的迷宫描述,它表示的就是上图中的迷宫)
特别注意:本题的测试数据非常大,请使用scanf输入,我不能保证使用cin能不超时.在本OJ上请使用Visual C++提交. 

Output

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

Sample Input

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

Sample Output

11

My Code

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

int dx[6] = {1, 0, -1, 0, 0, 0};
int dy[6] = {0, 1, 0, -1, 0, 0};
int dz[6] = {0, 0, 0, 0, 1, -1};
int map[55][55][55];
int v[55][55][55];
int a, b, c, t, T;
struct node
{
    int x, y, z, step;
} s;
bool check(node p)
{
    if (p.x < 0 || p.y < 0 || p.z < 0)
        return false;
    if (p.x >= a || p.y >= b || p.z >= c)
        return false;
    if (map[p.x][p.y][p.z])
        return false;
    if (v[p.x][p.y][p.z])
        return false;
    return true;
}
void bfs()
{
    memset(v, 0, sizeof(v));
    queue<node> q;
    node p, next;
    s.x = 0;
    s.y = 0;
    s.z = 0;
    s.step = 0;
    v[s.x][s.y][s.z] = 1;
    q.push(s);
    while (!q.empty())
    {
        p = q.front();
        q.pop();
        for (int i = 0; i < 6; i++)
        {
            next.x = p.x + dx[i];
            next.y = p.y + dy[i];
            next.z = p.z + dz[i];
            if (check(next))
            {
                next.step = p.step + 1;
                if (next.x == a - 1 && next.y == b - 1 && next.z == c - 1)
                {
                    printf("%d
", next.step);
                    return;
                }
                if (next.step > t)              //超时了就不用再试下去了
                    break;
                v[next.x][next.y][next.z] = 1;
                q.push(next);
            }
        }
    }
    printf("%d
", -1);
}
int main()
{
    scanf("%d", &T);
    while (T--)
    {
        scanf("%d%d%d%d", &a, &b, &c, &t);
        for (int i = 0; i < a; i++)
            for (int j = 0; j < b; j++)
                for (int k = 0; k < c; k++)
                    scanf("%d", &map[i][j][k]); //这题真的一定要用scanf
        if (map[a - 1][b - 1][c - 1] || a + b + c - 3 > t)
        {                                       //如果门前是墙,或者直接冲过去都来不及
            printf("%d
", -1);                 //那就不用试了
            continue;
        }
        bfs();
    }

    return 0;
}