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");    }}