[hdu] A计划 （BFS）

A计划
Time Limit : 3000/1000ms (Java/Other)   Memory Limit : 32768/32768K (Java/Other)

Total Submission(s) : 10   Accepted Submission(s) : 7

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
 
[cpp] 
// 664.cpp : 定义控制台应用程序的入口点。 
// 
 
#include "stdafx.h" 
 
 
#include<cstdio> 
#include<cstring> 
#include<queue> 
#define INF 0x3f3f3f3f 
#define MEM(arr,w) memset(arr,w,sizeof(arr)); 
using namespace std; 
struct node 
{ 
    int z,x,y; 
}st,ed,ss,tt; 
int dir[4][2]={0,1,1,0,0,-1,-1,0}; 
char map[2][12][12]; 
bool visit[2][12][12]; 
int step[2][12][12]; 
int n,m,t,MAXT; 
void BFS() 
{ 
    queue<node> Q; 
    Q.push(st); 
    visit[st.z][st.x][st.y]=true; 
    while(!Q.empty()) 
    { 
        ss=Q.front(); 
        Q.pop(); 
        if(ss.z==ed.z&&ss.x==ed.x&&ss.y==ed.y) 
            if(step[ss.z][ss.x][ss.y]<MAXT) MAXT=step[ss.z][ss.x][ss.y]; 
        if(MAXT<=t) return ; 
        for(int i=0;i<4;i++) 
        { 
            tt.z=ss.z; 
            tt.x=ss.x+dir[i][0]; 
            tt.y=ss.y+dir[i][1]; 
            if(tt.x>=0&&tt.x<n&&tt.y>=0&&tt.y<m&&map[tt.z][tt.x][tt.y]!='*'&&!visit[tt.z][tt.x][tt.y]) 
            { 
                if(map[tt.z][tt.x][tt.y]=='#')  
                { 
                    tt.z=1-tt.z; 
                    if(map[tt.z][tt.x][tt.y]=='*'||map[tt.z][tt.x][tt.y]=='#') continue; 
                     
                } 
                step[tt.z][tt.x][tt.y]=step[ss.z][ss.x][ss.y]+1; 
                if(step[tt.z][tt.x][tt.y]>t) continue; 
                Q.push(tt); 
                visit[tt.z][tt.x][tt.y]=true;    
                         
            } 
        } 
    } 
    return ; 
 
} 
int main() 
{ 
    int T,i,j,k; 
    scanf("%d",&T); 
    while(T--) 
    { 
        scanf("%d%d%d",&n,&m,&t); 
        for(i=0;i<2;i++) 
        { 
            for(j=0;j<n;j++) 
            { 
               scanf("%s",map[i][j]); 
               for(k=0;k<m;k++) 
               { 
                 //  if(map[i][j][k]=='S') st.z=i,st.x=j,st.y=k; 
                   st.z=0,st.x=0,st.y=0; 
                   if(map[i][j][k]=='P') ed.z=i,ed.x=j,ed.y=k; 
               } 
            }                  
        } 
        MAXT=INF; 
        MEM(visit,false); 
        MEM(step,0); 
        BFS(); www.2cto.com
        if(MAXT<=t) puts("YES"); 
        else puts("NO"); 
    } 
}