[kuangbin带你飞]专题二 搜索进阶 I

I - A计划

可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后，而今，不幸的她再一次面临生命的考验。魔王已经发出消息说将在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

写了好长时间过不去….最后加了个else就过去了….

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int INF=0x3f3f3f3f;
int n,m,t;
char ans[2][15][15];
bool book[2][15][15];
int cnt[4][3]={0,0,1,0,0,-1,0,1,0,0,-1,0};
struct Node {
    int x,y,z,num;
    bool operator < (const Node a) const {
        return num>a.num;
    }
};

bool check(int x,int y,int z) {
    if(x<0||x>1) return false;
    if(y<0||y>=n) return false;
    if(z<0||z>=m) return false;
    return true;
}

int bfs() {
    book[0][0][0]=true;
    Node now,next;
    now.x=now.y=now.z=now.num=0;
    priority_queue<Node>Q;
    Q.push(now);
    while(!Q.empty()) {
        now=Q.top();
        Q.pop();
        if(ans[now.x][now.y][now.z]=='P') return now.num;
        for(int i=0;i<4;i++) {
            int dx=now.x+cnt[i][0],dy=now.y+cnt[i][1],dz=now.z+cnt[i][2];
            if(check(dx,dy,dz)&&ans[dx][dy][dz]!='*'&&book[dx][dy][dz]==false) {
                if(ans[dx][dy][dz]=='#') {
                    if(dx==0&&ans[dx+1][dy][dz]!='*'&&ans[dx+1][dy][dz]!='#'&&book[dx+1][dy][dz]==false) {
                        next.x=dx+1,next.y=dy,next.z=dz,next.num=now.num+1;
                        Q.push(next);
                        book[dx+1][dy][dz]=true;
                    }
                   if(dx==1&&ans[dx-1][dy][dz]!='*'&&ans[dx-1][dy][dz]!='#'&&book[dx-1][dy][dz]==false) {
                        next.x=dx-1,next.y=dy,next.z=dz,next.num=now.num+1;
                        Q.push(next);
                        book[dx-1][dy][dz]=true;
                    }
                }
                else {
                        next.x=dx,next.y=dy,next.z=dz,next.num=now.num+1;
                        Q.push(next);
                        book[dx][dy][dz]=true;
                }
            }
        }
    }
    return -1;
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    int C;
    cin>>C;
    for(int k=0;k<C;k++) {
        cin>>n>>m>>t;
        for(int i=0;i<2;i++) {
            for(int j=0;j<n;j++) {
                for(int x=0;x<m;x++) {
                    cin>>ans[i][j][x];
                }
            }
        }
        memset(book,false,sizeof(book));
        int sum=bfs();
        if(sum<=t&&sum!=-1) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}