HDU 2102 A计划( BFS广度优先搜索)

http://acm.hdu.edu.cn/showproblem.php?pid=2102
A计划

Time Limit: 3000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 16708 Accepted Submission(s): 4197

Problem Description
可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后，而今，不幸的她再一次面临生命的考验。魔王已经发出消息说将在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表示迷宫的第一层的布置情况，后N*M表示迷宫第二层的布置情况。

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

Sample Input

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

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

Sample Output
YES

解题思路：跟平常的搜索没有什么太大的区别，个人的难点在于维度的处理和对题目的理解，题目中有个坑点：从传送机传送过去是不需要花费时间的。下面的东西就好处理了。
下面是AC代码：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;

struct node
{
    int x,y,z,step;
};
int dir[4][2]= {{0,1},{0,-1},{1,0},{-1,0}},book[2][15][15],n,m;
char a[2][15][15];
int ex,ey,ez;
int bfs(int sx,int sy,int sz,int ex,int ey,int ez)
{
    memset(book,0,sizeof(book));
    queue<node>q;
    node fr,ne;
    fr.x=sx,fr.y=sy,fr.z=sz,fr.step=0;
    q.push(fr);
    while(!q.empty())
    {
        fr=q.front();
        q.pop();
        if(fr.x==ex&&fr.y==ey&&fr.z==ez)
        {
            return fr.step;
        }
        for(int i=0; i<4; i++)
        {
            ne.x=fr.x+dir[i][0];
            ne.y=fr.y+dir[i][1];
            ne.z=fr.z;
            ne.step=fr.step+1;
            if(a[ne.z][ne.x][ne.y]!='*'&&book[ne.z][ne.x][ne.y]==0&&ne.x>=0&&ne.x<n&&ne.y>=0&&ne.y<m)
            {
                book[ne.z][ne.x][ne.y]=1;
                if(a[ne.z][ne.x][ne.y]=='#')
                {
                    if(ne.z==0&&book[1][ne.x][ne.y]==0)
                    {
                        ne.z=1;
                        if(a[ne.z][ne.x][ne.y]=='*'||a[ne.z][ne.x][ne.y]=='#')
                            continue;
                        book[ne.z][ne.x][ne.y]=1;
                        q.push(ne);
                    }
                    else if(book[0][ne.x][ne.y]==0)
                    {
                        ne.z=0;
                        if(a[ne.z][ne.x][ne.y]=='*'||a[ne.z][ne.x][ne.y]=='#')
                            continue;
                        book[ne.z][ne.x][ne.y]=1;
                        q.push(ne);
                    }
                }
                else
                {
                    q.push(ne);
                }
            }
        }
    }
    return -1;
}

int main()
{
    int t,time;
    scanf("%d",&t);
    while(t--)
    {
        memset(a,'\0',sizeof(a));
        scanf("%d%d%d",&n,&m,&time);
        for(int i=0; i<2; i++)
        {
            for(int j=0; j<n; j++)
            {
                scanf("%s",&a[i][j]);
                for(int k=0; k<m; k++)
                {
                    if(a[i][j][k]=='P')
                    {
                        ex=j,ey=k,ez=i;
                    }
                }
            }
        }
        int re=bfs(0,0,0,ex,ey,ez);
        if(re==-1||re>time)
        {
            printf("NO\n");
        }
        else
        {
            printf("YES\n");
        }
    }
    return 0;
}