胜利大逃亡--hdu1253

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=1253

广搜题

#include <stdio.h>
#include <string.h>
#include <queue>
#include <algorithm>
#define inf 0x6ffffff
#define N 55
using namespace std;

int maps[N][N][N],vis[N][N][N];

int dir[6][3]={ {0,0,1},{0,1,0},{1,0,0},{0,0,-1},{0,-1,0},{-1,0,0} };

int A,B,C,t;

struct node
{
    int x,y,z,step;
};

bool judge(int x,int y,int z)
{
    return x<A&&x>=0&&y<B&&y>=0&&z<C&&z>=0&&maps[x][y][z]==0&&vis[x][y][z]==0;
}

int bfs(node s,node e)
{
    queue<node> Q;
    node q;
    memset(vis,0,sizeof(vis));
    vis[s.x][s.y][s.z]=1;
    Q.push(s);
    while(!Q.empty())
    {
        q=Q.front();
        Q.pop();
        if(q.x==e.x&&q.y==e.y&&q.z==e.z && q.step <=t)
            return q.step;

        if(q.step > t)
            return -1;

        for(int i=0;i<6;i++)
        {
            s.x=q.x+dir[i][0];
            s.y=q.y+dir[i][1];
            s.z=q.z+dir[i][2];
            if(judge(s.x, s.y, s.z))
            {
                vis[s.x][s.y][s.z]=1;
                s.step=q.step+1;
                Q.push(s);
            }
        }
    }
    return -1;

}

int main()
{
    int T,i,j,k,ans;
    node s,e;
    scanf("%d",&T);
    while(T--)
    {
        memset(maps,0,sizeof(maps));

        scanf("%d %d %d %d",&A,&B,&C,&t);
        for(i=0;i<A;i++)
        {
            for(j=0;j<B;j++)
            {
                for(k=0;k<C;k++)
                {
                    scanf("%d",&maps[i][j][k]);
                }
            }
        }
        s.x=s.y=s.z=s.step=0;
        e.x=A-1;e.y=B-1;e.z=C-1;

        ans=bfs(s,e);

        if(A+B+C-3>t)
            ans=-1;
        printf("%d\n",ans);
    }
    return 0;
}