hdu 2102 A计划 BFS

Problem 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
  
  

 

其实还是很简单的一题，只不过地图变成了三维的。

当初写搓了就是没处理好时空传输。后来用一个deal（）函数感觉清晰了很多。。

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>

using namespace std;

int vis[2][15][15];
char map[2][15][15];
int n,m,time;
int dir[][2]={{0,1},{0,-1},{1,0},{-1,0}};

struct node{
    int x,y,step;
    int c;
};

void deal()
{
    for(int i=0;i<n;++i)
    for(int j=0;j<m;++j){
        char &a=map[0][i][j],&b=map[1][i][j];
        if(a=='#' && b=='#')    a=b='*';
        else if((a=='#' && b=='*') || (a=='*' && b=='#'))
            a=b='*';
    }
}
int bfs()
{
    queue<node> que;
    node s1,s2;
    memset(vis,0,sizeof(vis));
    s1.c=0;
    s1.x=0;    
    s1.y=0;
    s1.step=0;
    que.push(s1);
    vis[0][0][0]=1;
    while(!que.empty()){
        s1=que.front();
        que.pop();
        for(int i=0;i<4;++i){
            s2.x=s1.x+dir[i][0];
            s2.y=s1.y+dir[i][1];
            s2.c=s1.c;
            if(s2.x<0 || s2.x>=n || s2.y<0 || s2.y>=m)    continue;
            if(map[s2.c][s2.x][s2.y]=='*' || vis[s2.c][s2.x][s2.y])    continue;
            if(map[s2.c][s2.x][s2.y]=='#'){
                s2.c^=1;
                if(vis[s2.c][s2.x][s2.y])    continue;
            }
            if(map[s2.c][s2.x][s2.y]=='P')    return s1.step+1;
            s2.step=s1.step+1;
            vis[s2.c][s2.x][s2.y]=1;
            que.push(s2);
        }
    }
    return -1;
}
int main()
{
    int i,j,k,t;
    cin>>t;
    while(t--){
        scanf("%d%d%d",&n,&m,&time);
        for(k=0;k<2;++k){
            for(i=0;i<n;++i)
                scanf("%s",map[k][i]);
            if(!k)    getchar();
        }
        deal();
        int ans=bfs();
        //cout<<ans<<endl;
        if(ans==-1 || ans>time)    puts("NO");
        else puts("YES");
    }
    return 0;
}