A计划 HDU - 2102(三维BFS)

可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后，而今，不幸的她再一次面临生命的考验。魔王已经发出消息说将在T时刻吃掉公主，因为他听信谣言说吃公主的肉也能长生不老。年迈的国王正是心急如焚，告招天下勇士来拯救公主。不过公主早已习以为常，她深信智勇的骑士LJ肯定能将她救出。
现据密探所报，公主被关在一个两层的迷宫里，迷宫的入口是S（0，0，0），公主的位置用P表示，时空传输机用#表示，墙用*表示，平地用.表示。骑士们一进入时空传输机就会被转到另一层的相对位置，但如果被转到的位置是墙的话，那骑士们就会被撞死。骑士们在一层中只能前后左右移动，每移动一格花1时刻。层间的移动只能通过时空传输机，且不需要任何时间。

Input
输入的第一行C表示共有C个测试数据，每个测试数据的前一行有三个整数N，M，T。 N，M迷宫的大小N * M（1 <= N,M <=10)。T如上所意。 接下去的前N * M表示迷宫的第一层的布置情况，后NM表示迷宫第二层的布置情况。*

Output

如果骑士们能够在T时刻能找到公主就输出“YES”，否则输出“NO”。

Sample Input

1
5 5 14
S*#*.
.#…
…
****.
…#.

….P
#.…
**…
….
*.#…

Sample Output

YES

思路：

BFS

两层迷宫，三维数组

注意#时空传输机：骑士们一进入时空传输机就会被转到另一层的相对位置，但如果被转到的位置是墙的话，那骑士们就会被撞死。（到时空传输机的位置必转到另外一层）

剩下的就是模板，多注意细节即可

代码：

#include<stdio.h>
#include<queue>
#include<string.h>
#include<algorithm>
using namespace std;
char a[3][15][15];
int book[3][15][15];
int dir[4][2]={0,1,0,-1,1,0,-1,0};
int n,m,t;
int ex,ey,ez,f;
struct node
{
    int x,y,z,s;
}u,v;
void bfs()
{
    queue<node>q;
    u.x=0;
    u.y=0;
    u.z=0;
    u.s=0;
    book[0][0][0]=1;
    q.push(u);
    f=0;
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        if(a[u.z][u.x][u.y]=='P')
        {
            if(u.s<=t)
            {
                f=1;
                break;
            }
        }
        for(int i=0;i<4;i++)
        {
            v.x=u.x+dir[i][0];
            v.y=u.y+dir[i][1];
            v.z=u.z;
            if(v.x>=0&&v.x<n&&v.y>=0&&v.y<m&&a[v.z][v.x][v.y]!='*'&&!book[v.z][v.x][v.y])
            {
                if(a[v.z][v.x][v.y]=='#')
                {
                    v.z=1-v.z;
                    if(a[v.z][v.x][v.y]=='*'||book[v.z][v.x][v.y]||a[v.z][v.x][v.y]=='#')
                    {
                        continue;
                    }
                    book[v.z][v.x][v.y]=1;
                    book[1-v.z][v.x][v.y]=1;
                    v.s=u.s+1;
                    q.push(v);
                }
                else if(a[v.z][v.x][v.y]!='#')
                {
                     book[v.z][v.x][v.y]=1;
                     v.s=u.s+1;
                     q.push(v);
                }
            }
        }
    }
}
int main()
{
    int c;
    scanf("%d",&c);
    while(c--)
    {
        f=0;
        memset(book,0,sizeof(book));
        scanf("%d %d %d",&n,&m,&t);
        for(int k=0;k<=1;k++)
        {
            for(int i=0;i<n;i++)
            {
                scanf("%s",a[k][i]);
                for(int j=0;j<m;j++)
                {
                    if(a[k][i][j]=='P')
                    {
                        ex=i;
                        ey=j;
                        ez=k;
                    }
                }
            }
        }
        bfs();
        if(!f)
            printf("NO\n");
        else
            printf("YES\n");
    }
    return 0;
}