HDU 2102 A计划

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2102

BFS的迷宫问题，和一般的BFS一样。

注意：1.如果#的对面是'*',那么‘#’就要变成‘*’。

            2.如果#的对面是'#',那么‘#’就要变成‘*’。

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
struct node
{
    int x,y,z;
    int time;
};
char map[2][15][15];
int n,m,t,vis[15][15][2];
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};

void BFS(node &vs,node &vd)
{
    int flag=0;
    queue<node>fuck;
    node vw,vn;
    vn.x=vs.x;
    vn.y=vs.y;
    vn.z=vs.z;
    vn.time=0;
    vis[vn.x][vn.y][vn.z]=1;
    fuck.push(vn);
    while(!fuck.empty())
    {
        int i;
        vn=fuck.front();
        fuck.pop();
        if(vn.time>t)
        {
            printf("NO\n");
            return;
        }
        if(vn.x==vd.x&&vn.y==vd.y&&vn.z==vd.z)
        {
            if(vn.time<=t)
            {
                flag=1;break;
            }
        }
        else
        {
            if(map[vn.z][vn.x][vn.y]!='#')
            {
                for(i=0;i<4;i++)
                {
                    vw.x=vn.x+dx[i];
                    vw.y=vn.y+dy[i];
                    vw.z=vn.z;
                    vw.time=vn.time+1;
                    if(!vis[vw.x][vw.y][vw.z]&&map[vw.z][vw.x][vw.y]!='*'&&vw.x>=0&&vw.y>=0&&vw.z>=0&&vw.x<n&&vw.y<m&&vw.z<2)
                    {
                        fuck.push(vw);
                        vis[vw.x][vw.y][vw.z]=1;
                        if(vw.x==vd.x&&vw.y==vd.y&&vw.z==vd.z)
                        {
                            if(vw.time<=t)
                            {
                                flag=1;break;
                            }
                        }
                    }
                }
            }
            else if(map[vn.z][vn.x][vn.y]=='#')
            {
                    vw.x=vn.x;
                    vw.y=vn.y;
                    vw.z=1-vn.z;
                    vw.time=vn.time;
                    fuck.push(vw);
                    vis[vw.x][vw.y][vw.z]=1;
                    if(vw.x==vd.x&&vw.y==vd.y&&vw.z==vd.z)
                    {
                        if(vn.time<=t)
                        {
                                flag=1;break;
                        }
                    }
            }
        }
        if(flag)
          break;
    }
    if(flag)
      printf("YES\n");
    else
     printf("NO\n");
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int i,j,k;
        memset(vis,0,sizeof(vis));
        scanf("%d%d%d",&n,&m,&t);
        node vs,vd;
        for(i=0;i<n;i++)
         scanf("%s",map[0][i]);

        for(i=0;i<n;i++)
         scanf("%s",map[1][i]);

         for(i=0;i<2;i++)
          for(j=0;j<n;j++)
           for(k=0;k<m;k++)
              {
                  if(map[i][j][k]=='S')
                  {
                      vs.x=j;vs.y=k;vs.z=i;
                  }
                  else if(map[i][j][k]=='P')
                  {
                      vd.x=j;vd.y=k;vd.z=i;
                  }
                  else if(map[i][j][k]=='#'&&(map[1-i][j][k]=='*'||map[1-i][j][k]=='#'))
                  {
                      map[i][j][k]='*';
                      map[1-i][j][k]='*';
                  }
              }
            BFS(vs,vd);
    }
}