A计划 (HDU - 2102)

可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后，而今，不幸的她再一次面临生命的考验。魔王已经发出消息说将在T时刻吃掉公主，因为他听信谣言说吃公主的肉也能长生不老。年迈的国王正是心急如焚，告招天下勇士来拯救公主。不过公主早已习以为常，她深信智勇的骑士LJ肯定能将她救出。
现据密探所报，公主被关在一个两层的迷宫里，迷宫的入口是S（0，0，0），公主的位置用P表示，时空传输机用#表示，墙用表示，平地用.表示。骑士们一进入时空传输机就会被转到另一层的相对位置，但如果被转到的位置是墙的话，那骑士们就会被撞死。骑士们在一层中只能前后左右移动，每移动一格花1时刻。层间的移动只能通过时空传输机，且不需要任何时间。
Input
输入的第一行C表示共有C个测试数据，每个测试数据的前一行有三个整数N，M，T。 N，M迷宫的大小NM（1 <= N,M <=10)。T如上所意。接下去的前NM表示迷宫的第一层的布置情况，后NM表示迷宫第二层的布置情况。
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;

int l,r,c;
struct node
{
    int x,y,z,step;
    node()
    {
        step=0;
    }
    node(int a,int b,int c,int d)
    {
        x=a,y=b,z=c,step=d;
    }
}S,e,temp;
char s[15][15][5];
int X[6]={1,-1,0,0,0,0};
int Y[6]={0,0,1,-1,0,0};
int Z[6]={0,0,0,0,-1,1};
bool visit[15][15][5];

bool check(int x,int y,int z)
{
    if(s[x][y][z]=='*'||x<1||x>l) return false;
    if(y>r||y<1||visit[x][y][z]) return false;
    return true;
}

int solve()
{
    visit[S.x][S.y][S.z]=true;
    queue<node> q;
    q.push(S);
    while(!q.empty())
    {
        temp=q.front();
        q.pop();
        if(temp.x==e.x&&temp.y==e.y&&temp.z==e.z) return temp.step;
        for(int i=0;i<4;i++)
        {
            int nx=temp.x+X[i];
            int ny=temp.y+Y[i];
            int nz=temp.z;
            if(!check(nx,ny,nz)) continue;
            visit[nx][ny][nz]=true;
            if(s[nx][ny][nz]=='#')
            {
                nz=!temp.z;///要过check判断，才能判断过不过
                ///注意上下层都为#的情况
                if(!check(nx,ny,nz)||s[nx][ny][nz]=='#')
                    continue;
                visit[nx][ny][nz]=true;
            }
            q.push({nx,ny,nz,temp.step+1});
        }
    }
    return -1;
}

int main()
{
    int t;cin>>t;
    while(t--)
    {
        cin>>l>>r>>c;
        memset(visit,0,sizeof(visit));
        memset(s,0,sizeof(s));
        for(int k=0;k<=1;k++)
            for(int i=1;i<=l;i++)
                for(int j=1;j<=r;j++)
                {
                    cin>>s[i][j][k];
                    if(s[i][j][k]=='S') S.x=i,S.y=j,S.z=k;
                    if(s[i][j][k]=='P') e.x=i,e.y=j,e.z=k;
                }
        int ans=solve();
        if(ans>c||ans==-1) cout<<"NO"<<endl;///-1忘了判
        else cout<<"YES"<<endl;
    }
    return 0;
}
/*
1
5 5 14
S*#*.
.#...
.....
****.
...#.

..*.P
#**..
***..
...*.
*.##.
*/