HDU 2102--A计划【BFS】

A计划

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

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 <cstdio>
#include <iostream>
#include <algorithm>
#include <queue>
#include <cstring>
#define maxn 1010
using namespace std;

char map[2][maxn][maxn];
bool vis[2][maxn][maxn];
int dx[4]={0,1,0,-1};
int dy[4]={1,0,-1,0};
int n,m,T,sx,sy,sz;

struct node {
    int x,y,z,t;
};

bool check(node a){
    if(a.x<0||a.x>=n||a.y<0||a.y>=m) return 0;
    if(map[a.z][a.x][a.y]=='#'||map[a.z][a.x][a.y]=='*') return 0;
    if(vis[a.z][a.x][a.y]) return 0;
    if(a.t>T) return 0;
    return 1;
}
bool bfs(){
    node st,ed;
    st.x=st.y=st.z=st.t=0;
    queue<node>q;
    vis[st.z][st.x][st.y]=1;
    q.push(st);
    while(!q.empty()){
        st=q.front();
        q.pop();
        for(int i=0;i<4;++i){
            ed=st;
            ed.x+=dx[i];
            ed.y+=dy[i];
            ed.t++;
            if(map[ed.z][ed.x][ed.y]=='#'){
                ed.z=!(ed.z);
            }
            if(check(ed)){
                if(map[ed.z][ed.x][ed.y]=='P') return 1;
                vis[ed.z][ed.x][ed.y]=1;
                q.push(ed);
            }
        }
    }
    return 0;
}
int main (){
    int t,i,j,k;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d%d",&n,&m,&T);
        for(k=0;k<2;++k)
            for(i=0;i<n;++i)
                    scanf("%s",map[k][i]);

        memset(vis,0,sizeof(vis));
        if(bfs()) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}