HDU2102: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

.*..

*..
…*.
*.#..

Sample Output

YES

#include<bits/stdc++.h>
using namespace std;
struct node{
   int x,y,z;
   int t;
}p;
int n,m,t;
char mp[15][15][15];
int vis[15][15][15];
bool check(int x,int y,int z)
{
    if(x<0||x==n||y<0||y==m)
        return false;
    if(mp[z][x][y]=='#') {vis[z][x][y]=1;z=!z;
       if(mp[z][x][y]=='#') return false;
    }
    if(vis[z][x][y]||mp[z][x][y]=='*')
        return false;
    return true;
}

int dis[4][2]={{0,1},{1,0},{-1,0},{0,-1}};

void bfs(int x,int y,int z)
{
    memset(vis,0,sizeof(vis));
    node nd;
    nd.x=x,nd.y=y,nd.z=z,nd.t=0;
    vis[z][x][y]=1;
    queue<node> q;
    q.push(nd);
    while(!q.empty())
    {
        nd=q.front();q.pop();
//        cout<<nd.t<<' '<<t<<endl;
//        cout<<"x="<<nd.x<<endl;
//        cout<<"y="<<nd.y<<endl;
//        cout<<"z="<<nd.z<<endl;
        if(nd.t>t) break;
        if(mp[nd.z][nd.x][nd.y]=='P')
        {
            printf("YES\n");
            return;
        }
        for(int i=0;i<4;i++)
        {
            if(check(nd.x+dis[i][0],nd.y+dis[i][1],nd.z))
            {
                node tt;
                tt.x=nd.x+dis[i][0];
                tt.y=nd.y+dis[i][1];
                tt.z=nd.z;
                if(mp[tt.z][tt.x][tt.y]=='#')
                tt.z=(tt.z+1)%2;
                tt.t=nd.t+1;
                vis[tt.z][tt.x][tt.y]=1;
                q.push(tt);
            }
        }
    }
    printf("NO\n");
}


int main()
{
    int c;
    cin>>c;
    while(c--)
    {
        cin>>n>>m>>t;
        for(int z=0;z<2;z++)
         for(int i=0;i<n;i++)
         cin>>mp[z][i];
        for(int z=0;z<2;z++)
        {
            for(int i=0;i<n;i++)
            {
                for(int j=0;j<m;j++)
                {
                    if(mp[z][i][j]=='S')
                    {
                        bfs(i,j,z);
                        break;
                    }
                }
            }
        }
    }
}