hdu2102: A计划

hdu2102: http://acm.hdu.edu.cn/showproblem.php?pid=2102 
题意：迷宫的入口是S（0，0，0），公主的位置用P表示，时空传输机用#表示，墙用*表示，平地用.表示。骑士们一进入时空传输机就会被转到另一层的相对位置，但如果被转到的位置是墙   的话，那骑士们就会被撞死。骑士们在一层中只能前后左右移动，每移动一格花1时刻。层间的移动只能通过时空传输机，且不需要任何时间。
code：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
char v[2][20][20];
int vis[2][20][20],ans[2][20][20],q[100][2];
int dx[]={-1,1,0,0};
int dy[]={0,0,-1,1};
const int inf=1<<29;
int main()
{
    int c,n,m,t,i,j,front,rear,u,k,w,a,b,x,y,nx,ny,d,e,f,k1,mx,my,o,p,r;
    scanf("%d",&c);
    while(c--)
    {
        scanf("%d%d%d",&n,&m,&t);
        memset(vis,0,sizeof(vis));
        for(i=0;i<2;i++)
            for(j=0;j<20;j++)
                for(p=0;p<20;p++)
                    ans[i][j][p]=inf;
        getchar();
        for(i=0;i<n;i++)
        {
            gets(v[0][i]);
            for(j=0;j<m;j++)
            {
               if(v[0][i][j]=='S')
               {
                   vis[0][i][j]=1;ans[0][i][j]=0;
                   a=m*i+j;k=0;
               }
                if(v[0][i][j]=='P')
                {
                    d=0;e=i;f=j;
                    b=m*i+j;
                }
            }
        }
        getchar();
        for(i=0;i<n;i++)
        {
            gets(v[1][i]);
            for(j=0;j<m;j++)
            {
                if(v[1][i][j]=='S')
                {
                    vis[1][i][j]=1;ans[1][i][j]=0;
                    a=m*i+j;k=1;
                }
                if(v[1][i][j]=='P')
                {
                    d=1;e=i;f=j;
                    b=m*i+j;
                }
            }
        }
        front=0;rear=0;
        q[rear++][0]=a;q[rear-1][1]=k;
        while(front<rear)
        {
            u=q[front++][0];
            k=q[front-1][1];
            x=u/m;y=u%m;
            for(i=0;i<4;i++)
            {
                nx=x+dx[i];ny=y+dy[i];
                if(nx>=0&&nx<n&&ny>=0&&ny<m&&!vis[k][nx][ny]&&v[k][nx][ny]!='*')
                {
                    vis[k][nx][ny]=1;
                    w=nx*m+ny;
                    if(k==0)
                        k1=1;
                    else
                        k1=0;
                    if(v[k][nx][ny]=='#'&&(v[k1][nx][ny]!='*')&&(v[k1][nx][ny]!='#'))  //注意考虑两层都为'#'的情况是不可取的
                    {
                        ans[k1][nx][ny]=ans[k1][nx][ny]>ans[k][x][y]+1?ans[k][x][y]+1:ans[k1][nx][ny];
                        q[rear++][0]=w;
                        q[rear-1][1]=k1;
                    }
                    else if(!(v[k][nx][ny]=='#'))
                    {
                        ans[k][nx][ny]=ans[k][nx][ny]>ans[k][x][y]+1?ans[k][x][y]+1:ans[k][nx][ny];
                        q[rear++][0]=w;
                        q[rear-1][1]=k;
                    }
                }
            
            }
        }
        if(ans[d][e][f]>t)
            printf("NO\n");
        else
            printf("YES\n");
    }
}
/*
Sample Input
1
5 5 14
S*#*.
.#...
.....
****.
...#.

..*.P
#.*..
***..
...*.
*.#..
Sample Output
YES
*/