BFS入门-A计划

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<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
struct node
{
int x;
int y;
int z;
}S,E,Ne;
int main()
{
int C,N,M,T,dir[4][2]={1,0,-1,0,0,1,0,-1};
char one[20][20],two[20][20];
scanf("%d",&C);
while(C--)
{
scanf("%d %d %d",&N,&M,&T); 
   int i,j,o[20][20]={1},t[20][20]={0};
   S.x=0;
   S.y=0;
   S.z=0;
   memset(one,0,sizeof(one));
        memset(two,0,sizeof(two));
   for(i=0;i<N;i++)scanf("%s",one[i]);
   for(i=0;i<N;i++)scanf("%s",two[i]);
   int flag=0;
   for(i=0;i<N;i++)
   {
    for(j=0;j<M;j++)
    {
    if(one[i][j]=='P')
    {
    E.x=i;
    E.y=j;
    E.z=0;
    flag=1;
    }
    if(flag)break;
    if(two[i][j]=='P')
    {
    E.x=i;
    E.y=j;
    E.z=1;
    flag=1;
    }
    if(flag)break;
   }
   if(flag)break;
    }
    flag=0;
    queue<node> qu;
    int count=0;
    qu.push(S);
    while(count<T)//时间限制 
    {
    count++;
    int l=qu.size();
    while(l--)
    {
    S=qu.front();
    qu.pop();
    for(i=0;i<4;i++)
    {
    Ne.x=S.x+dir[i][0];
    Ne.y=S.y+dir[i][1];
    Ne.z=S.z;
    if(Ne.z==0)//在第一层 
    {
    if(one[Ne.x][Ne.y]=='P')
    {
    printf("YES\n");
    flag=1;
    }
else if(one[Ne.x][Ne.y]=='.'&&o[Ne.x][Ne.y]==0&&Ne.x>=0&&Ne.x<N&&&Ne.y>=0&&Ne.y<M)
    {
       o[Ne.x][Ne.y]=1;
    qu.push(Ne);
    }
    else if(one[Ne.x][Ne.y]=='#'&&o[Ne.x][Ne.y]==0&&Ne.x>=0&&Ne.x<N&&&Ne.y>=0&&Ne.y<M)//跳转 
{
    if(two[Ne.x][Ne.y]=='P')
       {
      printf("YES\n");
      flag=1;
       }
else if(t[Ne.x][Ne.y]==0&&two[Ne.x][Ne.y]!='*'&&two[Ne.x][Ne.y]!='#')
    {
    Ne.z=1;
    qu.push(Ne);
    Ne.z=0;
    t[Ne.x][Ne.y]=1;
    }
    o[Ne.x][Ne.y]=1;
} 
    }
    else if(Ne.z==1)//在第二层 
    {
    if(two[Ne.x][Ne.y]=='P')
    {
    printf("YES\n");
    flag=1;
    }
else if(two[Ne.x][Ne.y]=='.'&&t[Ne.x][Ne.y]==0&&Ne.x>=0&&Ne.x<N&&&Ne.y>=0&&Ne.y<M)
    {
    t[Ne.x][Ne.y]=1;
    qu.push(Ne);
    }
    else if(two[Ne.x][Ne.y]=='#'&&t[Ne.x][Ne.y]==0&&Ne.x>=0&&Ne.x<N&&&Ne.y>=0&&Ne.y<M)//跳转 
{
    if(one[Ne.x][Ne.y]=='P')
       {
       printf("YES\n");
       flag=1;
       }
else if(o[Ne.x][Ne.y]==0&&one[Ne.x][Ne.y]!='*'&&one[Ne.x][Ne.y]!='#')
    {
    Ne.z=0;
    qu.push(Ne);
    o[Ne.x][Ne.y]=1;
    }
    t[Ne.x][Ne.y]=1;
} 
    }
    }
    if(flag)break;
    }
    if(flag)break;
   }
   if(!flag)printf("NO\n");
}
}