hdu 2102 A计划（bfs）

之前没有考虑到 两层都是 # 的情况 

 

#include<cstdio>

#include<cmath>

#include<stdio.h>

#include<string.h>

#include<math.h>

#include<iostream>

#include<algorithm>

#include<queue>

#include<stack>

#define mem(a,b) memset(a,b,sizeof(a))

using namespace std;

int n,m,coun,ok1;

char mat[5][20][20];

int op[4][2]={{1,0},{-1,0},{0,1},{0,-1}};

struct node{

  int t;

  int w,x,y;

};



bool ok(node next)

{

    if(next.x<0||next.x>=n||next.y<0||next.y>=m) return false;   

    if(mat[next.w][next.x][next.y]=='S'||mat[next.w][next.x][next.y]=='*') return false;

    return true;

}

void bfs()

{

    queue<node> q;

    node now,next;

    now.t=0;

    now.w=0;

    now.x=0;

    now.y=0;

    q.push(now);

    while(!q.empty())

    {

        now=q.front();

        q.pop();

        if(now.t>=coun) return;



        for(int i=0;i<4;i++)

        {

            next.x=now.x+op[i][0];

            next.y=now.y+op[i][1];

            next.w=now.w;

            if(ok(next))

            {

                next.t=now.t+1;

                if(mat[now.w][next.x][next.y]=='#') {next.w=1-now.w;}

                if(!ok(next)) continue;

            if(mat[next.w][next.x][next.y]=='P')

            {

                ok1=1;

                return ;

            }            

            if(mat[next.w][next.x][next.y]=='.')

            {

                mat[next.w][next.x][next.y]='S';

                q.push(next);

            }            

            }            

        }

    }

}

int main()

{

    int i,t;

    cin>>t;

    while(t--)

    {

        ok1=0;

        scanf("%d%d%d",&n,&m,&coun);

        memset(mat,0,sizeof(mat));

        for(i=0;i<n;i++)

        {

            scanf("%s",mat[0][i]);

        }

        for(i=0;i<n;i++)

        {

            scanf("%s",mat[1][i]);

        }

        bfs();

        if(ok1)

            printf("YES\n");

        else printf("NO\n");

    }

    return 0;

}