HDU - 2102 A计划（先处理地图，再进行广搜）

题目链接：https://vjudge.net/contest/351640#problem/G

Sample Input

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

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

Sample Output

YES

分析：
对于传送门，先考虑传送情况，对地图进行处理，再进行广搜
代码：

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int N=20;
int to[4][2]= {-1,0,0,-1,1,0,0,1};
char mp[2][N][N];
int n,m,t;
struct node
{
    int x,y,z,step;
};
int bfs(int x,int y,int z)
{
    mp[x][y][z]='*';
    node u,v;
    queue<node>q;
    u.x=x,u.y=y,u.z=z,u.step=0;
    q.push(u);
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        for(int i=0; i<4; i++)
        {
            v.x=u.x;/*在那一层先是不变的,除非遇到传送机*/
            v.y=u.y+to[i][0],v.z=u.z+to[i][1];
            v.step=u.step+1;
            if(v.y>=0&&v.y<n&&v.z>=0&&v.z<m&&mp[v.x][v.y][v.z]!='*')
            {
                if(mp[v.x][v.y][v.z]=='#')/*传送机的位置*/
                {
                    mp[v.x][v.y][v.z]='*';
                    if(v.x==1)/*传送到另一层*/
                        v.x=0;
                    else if(v.x==0)
                        v.x=1;
                }
                if(mp[v.x][v.y][v.z]=='P'&&v.step<=t)/*找到公主*/
                    return 1;
                mp[v.x][v.y][v.z]='*';/*正常的走到路的情况*/
                q.push(v);
            }
        }
    }
    return 0;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&m,&t);
        for(int k=0; k<2; k++)
        {
            for(int i=0; i<n; i++)
                scanf("%s",mp[k][i]);
        }
        for(int i=0; i<n; i++)/*传送情况*/
        {
            for(int j=0; j<m; j++)
            {
                if(mp[0][i][j]=='#'&&mp[1][i][j]=='*')
                    mp[0][i][j]='*';
                if(mp[0][i][j]=='*'&&mp[1][i][j]=='#')
                    mp[1][i][j]='*';
                if(mp[0][i][j]=='#'&&mp[1][i][j]=='#')
                {
                    mp[0][i][j]='*';
                    mp[1][i][j]='*';
                }
            }
        }
        if(bfs(0,0,0)==1)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}