A计划 HDU - 2102

题目链接：A计划 HDU - 2102

===================================================

A计划

Time Limit: 1000MS
Memory Limit: 32768 kB

Description

可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后，而今，不幸的她再一次面临生命的考验。魔王已经发出消息说将在T时刻吃掉公主，因为他听信谣言说吃公主的肉也能长生不老。年迈的国王正是心急如焚，告招天下勇士来拯救公主。不过公主早已习以为常，她深信智勇的骑士LJ肯定能将她救出。
现据密探所报，公主被关在一个两层的迷宫里，迷宫的入口是S（0，0，0），公主的位置用P表示，时空传输机用#表示，墙用*表示，平地用.表示。骑士们一进入时空传输机就会被转到另一层的相对位置，但如果被转到的位置是墙的话，那骑士们就会被撞死。骑士们在一层中只能前后左右移动，每移动一格花1时刻。层间的移动只能通过时空传输机，且不需要任何时间。

Input

输入的第一行C表示共有C个测试数据，每个测试数据的前一行有三个整数N，M，T。 N，M迷宫的大小NM（1 <= N,M <=10)。T如上所意。接下去的前NM表示迷宫的第一层的布置情况，后N*M表示迷宫第二层的布置情况。

Output

如果骑士们能够在T时刻能找到公主就输出“YES”，否则输出“NO”。

Sample Input

1
5 5 14
S*#*.
.#…
…
****.
…#.

….P
#.…
**…
….
*.#…

Sample Output

YES

===================================================

算法：bfs

难点：

• 在于题意中的判断，因为就两层还好，传送门是否可以通过？

===================================================

#include <iostream>
#include <queue>
#include <cstring>

using namespace std;

const int INF = 2e9;

int n,m,t;
int Px,Py,Pz;

int ditu[11][11][3];
bool vis[11][11][3];

int xx[] = {1,0,-1,0};
int yy[] = {0,1,0,-1};

struct node{
    int x,y,z,step;
    node(int a,int b,int c,int d):x(a),y(b),z(c),step(d) {};
};
queue<node> q;

void init(){
    memset(ditu,0,sizeof ditu);
    memset(vis,true,sizeof vis);
    while(!q.empty()) q.pop();
    cin>>n>>m>>t;
    for(int z=0;z<2;z++)
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++){
                char ch = getchar();while(ch==' '||ch=='\n') ch=getchar();
                if(ch=='*') ditu[i][j][z] = -1;
                else if(ch=='#') ditu[i][j][z] = 1;
                else if(ch=='S') q.push(node(i,j,z,0)),vis[i][j][z] = false;
                else if(ch=='P') Px = i,Py = j,Pz = z;
            }
}

bool check(int x,int y,int z){
    if(x<0||x>=n||y<0||y>=m||z<0||z>=2) return false;
    if(ditu[x][y][z]==-1) return false;
    if(ditu[x][y][z]==1&&(ditu[x][y][z^1]==-1||ditu[x][y][z^1]==1)) return false;
    return true;
}

bool bfs(){
    while(!q.empty()){
        node p = q.front();q.pop();
        if(p.step>t) continue;
        if(Px==p.x&&Py==p.y&&Pz==p.z) return true;
        for(int i=0;i<4;i++){
            int x = p.x + xx[i] , y = p.y + yy[i],z = p.z;
            if(!check(x,y,z)||!vis[x][y][z]) continue;
            vis[x][y][z] = false;if(ditu[x][y][z]==1) z^=1;vis[x][y][z] = false;
            q.push(node(x,y,z,p.step+1));
        }
    }
    return false;
}

int main()
{
    int _;cin>>_;
    while(_--){
        init();
        if(bfs()) puts("YES");
        else puts("NO");
    }
    return 0;
}