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
#**..
***..
...*.
*.##.
*/