HDU 1253 胜利大逃亡(BFS)

#include <cstdio>
#include <iostream>
#include <string>
#include <cstring>
#include <stack>
#include <queue>
#include <algorithm>
#include <cmath>
#include <map>
using namespace std;
//#define LOCAL
int mp[55][55][55];
bool vis[55][55][55];
int N,A,B,C,T;
struct node
{
    node(int a,int b,int c,int d)
    {x=a,y=b,z=c,time=d;}
    int x,y,z;
    int time;
};
int dir[6][3]={{1,0,0},{-1,0,0},{0,0,1},{0,0,-1},{0,1,0},{0,-1,0}};
bool check(int x,int y,int z)
{
    if(x>=0&&y>=0&&z>=0&&x<A&&y<B&&z<C&&mp[x][y][z]==0)
    return 1;
    return 0;
}
int bfs(int t)
{
    node st(0,0,0,0),ed(0,0,0,0);
    queue<node>q;
    while(!q.empty())q.pop();
    memset(vis,0,sizeof vis);
    vis[0][0][0]=1;
    q.push(st);
    while(!q.empty())
    {
        st=q.front();
        q.pop();
        if(st.time>t)return -1;
        if(st.x==A-1&&st.y==B-1&&st.z==C-1&&st.time<=T)return st.time;
        for(int i=0;i<6;i++)
        {
            ed.x=st.x+dir[i][0];
            ed.y=st.y+dir[i][1];
            ed.z=st.z+dir[i][2];
            if(check(ed.x,ed.y,ed.z)&&!vis[ed.x][ed.y][ed.z])
            {
                vis[ed.x][ed.y][ed.z]=true;
                ed.time=st.time+1;
                if(abs(ed.x-A+1)+abs(ed.y-B+1)+abs(ed.z-C+1)+ed.time>T)continue;
                q.push(ed);
            }
        }
    }
    return -1;
}
int main()
{
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
#endif // LOCAL
    //Start
    scanf("%d",&N);
    while(N--)
    {
        scanf("%d%d%d%d",&A,&B,&C,&T);
        memset(mp,0,sizeof mp);
        for(int i=0; i<A; i++)
        for(int j=0; j<B; j++)
        for(int k=0; k<C; k++)
        scanf("%d",&mp[i][j][k]);
        printf("%d\n",bfs(T));
    }
    return 0;
}