HDU 2102 A计划

/*要在T秒之内从S到P*/
/*二层迷宫，时空转换器*/
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
int n,m,T,ex,ey,ez;//e表示终点
char a[10+1][10+1][2+1];
bool vis[10+1][10+1][2+1];
struct node
{
    int xx,yy,zz,time;
}person;
bool check(int x,int y,int z)
{
    if(x>=1&&x<=n&&y>=1&&y<=m&&(z==1||z==2))//在界内
        if(a[x][y][z]!=‘*‘)//不为墙
            if(!vis[x][y][z])//未曾访问过
                return true;
    return false;
}
int rx[]={0,0,1,-1};
int ry[]={1,-1,0,0};
bool bfs()
{
    queue<node>Q;
    Q.push(person);
    node cur,next;
    while(!Q.empty())
    {
        cur=Q.front();
        Q.pop();
        vis[cur.xx][cur.yy][cur.zz]=true;
        if(cur.xx==ex&&cur.yy==ey&&cur.zz==ez&&cur.time<=T)
            return true;
        if(cur.time>T)
            break;
        if(a[cur.xx][cur.yy][cur.zz]==‘#‘)
        {
            next.xx=cur.xx;
            next.yy=cur.yy;
            if(cur.zz==1)next.zz=2;
            else next.zz=1;
            if(check(next.xx,next.yy,next.zz))
                Q.push(next);
        }
        else
        {
            for(int i=0;i<4;i++)
            {
                next.xx=cur.xx+rx[i];
                next.yy=cur.yy+ry[i];
                next.zz=cur.zz;
                next.time=cur.time+1;
                if(check(next.xx,next.yy,next.zz))
                    Q.push(next);
            }
        }
    }
    return false;
}
int main()
{
    int K;
    cin>>K;
    while(K--)
    {
        cin>>n>>m>>T;
        memset(vis,false,sizeof(vis));
        for(int z=1;z<=2;z++)
        {
            for(int x=1;x<=n;x++)
            {
                for(int y=1;y<=m;y++)
                {
                    cin>>a[x][y][z];
                    if(a[x][y][z]==‘S‘)
                    {
                        person.xx=x;
                        person.yy=y;
                        person.zz=z;
                        person.time=0;
                        vis[x][y][z]=true;
                    }//标记起点
                    else if(a[x][y][z]==‘P‘)
                    {
                        ex=x;
                        ey=y;
                        ez=z;
                    }//标记终点
                }
            }
        }
        //接下来对时空传输机进行预处理
        for(int x=1;x<=n;x++)
        {
            for(int y=1;y<=m;y++)
            {
                if(a[x][y][1]==‘#‘&&a[x][y][2]==‘#‘)
                {
                    a[x][y][1]=‘*‘;
                    a[x][y][2]=‘*‘;
                }
                else if(a[x][y][1]==‘#‘&&a[x][y][2]==‘*‘)
                {
                    a[x][y][1]=‘*‘;
                }
                else if(a[x][y][2]==‘#‘&&a[x][y][1]==‘*‘)
                {
                    a[x][y][2]=‘*‘;
                }
            }
        }
        if(bfs())
            cout<<"YES"<<endl;
        else 
            cout<<"NO"<<endl;
    }
    return 0;
}