hdu A计划 (BFS）

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

.*..

*..
…*.
*.#..
注意点：如果两层的相同位置都是传送门，那么就会无限传送；
思路：
1 因为被转到的位置是墙的话，骑士们就会被撞死，如果两层的相同位置都是传送门，那么就会无限传送，我们可以提前做预处理，把这两种情况的点都设为墙，就不会干涉到后面的搜索；
代码：

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
struct node{
   int t,x,y,z;
};
char map[15][15][5];
int mark[15][15][5];
int n,m,t,flag;
int dx,dy,dz;
int vis[4][2]={0,1,0,-1,1,0,-1,0};
int ok(int x,int y,int z)
{
    if(x>=1&&x<=n&&y>=1&&y<=m&&map[x][y][z]!='*')
       return 1;
    else
       return 0;
}
void bfs()
{
    node q,p;
    q.x=1,q.y=1,q.z=1,q.t=0;
    queue<node>que;
    que.push(q);
    while(que.size())
    {
        q=que.front();
        que.pop();
        if(q.t>t)
           return;
        if(q.x==dx&&q.y==dy&&q.z==dz)
        {
            flag=q.t;
            return;
        }
        for(int i=0;i<=3;i++)
        {
            p.x=q.x+vis[i][0];
            p.y=q.y+vis[i][1];
            if(ok(p.x,p.y,q.z)&&!mark[p.x][p.y][q.z])
            {
                mark[p.x][p.y][q.z]=1;
                if(map[p.x][p.y][q.z]=='#')
                {
                    if(q.z==1)
                       p.z=2;
                    else
                       p.z=1;
                    mark[p.x][p.y][p.z]=1;
                }
                else
                  p.z=q.z;
                p.t=q.t+1;
                que.push(p);
            }
        }
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        flag=0;
        scanf("%d%d%d",&n,&m,&t);
        getchar();
        for(int j=1;j<=2;j++)
        {
            for(int i=1;i<=n;i++)
            {
                for(int k=1;k<=m;k++)
                {
                    scanf("%c",&map[i][k][j]);
                    if(map[i][k][j]=='P')
                       dx=i,dy=k,dz=j;
                }
                getchar();
            }
            getchar();
        }
        for(int i=1;i<=n;i++)
           for(int j=1;j<=m;j++)
           {
               if(map[i][j][2]=='#'&&map[i][j][1]=='#')
                  map[i][j][1]='*',map[i][j][2]='*';
               if(map[i][j][2]=='#'&&map[i][j][1]=='*')
                  map[i][j][2]='*';
               if(map[i][j][1]=='#'&&map[i][j][2]=='*')
                  map[i][j][1]='*';
           }
        memset(mark,0,sizeof(mark));
        mark[1][1][1]=1;
        bfs();
        if(flag)
           printf("YES\n");
        else
           printf("NO\n");

    }
}