BFS_4

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

中文題就不解釋了，這題和普通的bfs相比無非就是多了一層，也就是三維的空間，一般來說只要開一個三維的數組就可以了。但是當初我並沒有考慮三維數組，因爲只有兩層，所以我就直接分情況了

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<string>
#include<queue>

using namespace std;

const int Maxn = 205;
struct point
{
  int x;
  int y;
  int level;
};

char pos1[Maxn][Maxn];//定義位置數組；
char pos2[Maxn][Maxn];

int a[Maxn][Maxn];//定義標記數組
int b[Maxn][Maxn];

/*4個前進方向*/
int X[4]={0,0,-1,1};
int Y[4]={1,-1,0,0};
int fi,fj,li,lj;
int n,m,T,ans=0;
int Count=0;
//bool flag;
queue<point>que;

point start,End;

void bfs()
{

    start.level=1;
    start.x=0;
    start.y=0;
    End.x=li;
    End.y=lj;
    que.push(start);
    a[start.x][start.y]=1;
    point temp,next;
    while(!que.empty())
    {
        ans++;
        temp=que.front();
        que.pop();
        int k;
        for(k=0;k<4;k++)
        {
            next.x=temp.x+X[k];
            next.y=temp.y+Y[k];
            if(next.level==1)
            {
                if(next.x>=0&&next.y>=0&&next.x<n&&next.y<m&&a[next.x][next.y]==0)
                {
                    a[next.x][next.y]=1;
                    if(pos1[next.x][next.y]=='.')  que.push(next);
                    else if(pos1[next.x][next.y]=='*')    continue;
                    else if(pos1[next.x][next.y]=='#')
                    {
                        next.level=2;
                        if(pos2[next.x][next.y]=='*')   continue;
                        else    que.push(next);
                    }
                }
                else  continue;
            }
            if(next.level==2)
            {
                if(next.x>=0&&next.y>=0&&next.x<n&&next.y<m&&b[next.x][next.y]==0)
                {
                    b[next.x][next.y]=1;
                    if(next.x == End.x && next.y == End.y)  return;
                    else if(pos2[next.x][next.y]=='.')  que.push(next);
                    else if(pos2[next.x][next.y]=='*')    continue;
                    else
                    {
                        next.level=1;
                        if(pos2[next.x][next.y]=='*')  continue;
                        else    que.push(next);
                    }
                }
                else  continue;
            }
        }
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&T);
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        while(!que.empty())
        que.pop();
        for(int i=0;i<n;i++)  scanf("%s",pos1[i]);
        for(int i=0;i<n;i++)  scanf("%s",pos2[i]);
        for(int i=0;i<n;i++)
        {
          for(int j=0;j<m;j++)
          {
            if(pos2[i][j]=='P')
            {
              fi=i;fj=j;bfs();
              break;
            }
          }
        }
        printf("%d",ans);
        if(ans<T)  printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}


/*
1
5 5 14
.*#*.
.#S..
.....
****.
...#.

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