hdu 1728 逃离迷宫 bfs

#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;

struct node{
int x,y,dir,step;
friend bool operator <(node aa, node bb)
{
return aa.step>bb.step;
}
}a;

int n,m,k,x1,y1,x2,y2;
char map[105][105];
int dp[105][105][4];
int d[4][2]={0,1,1,0,0,-1,-1,0};

void init()
{
int i;
for (i=0;i<n;i++)scanf("%s",map[i]);
scanf("%d%d%d%d%d",&k,&y1,&x1,&y2,&x2);
x1-=1;x2-=1;y1-=1;y2-=1;
a.dir=-1,a.step=0,a.x=x1,a.y=y1;
memset(dp,0,sizeof(dp));
}

void bfs()
{
int i,tx,ty;
priority_queue<node> q;
q.push(a);
while (!q.empty())
{
a=q.top();q.pop();
for (i=0;i<4;i++)
{
tx=a.x+d[i][0];
ty=a.y+d[i][1];
while (tx>=0&&ty>=0&&tx<n&&ty<m&&map[tx][ty]=='.'&&!dp[tx][ty][i])
{
node b;
b.x=tx;b.y=ty;b.dir=i;b.step=a.step;
if(i!=a.dir)b.step++;
if(b.step>k+1)break;
q.push(b);
dp[tx][ty][i]=1;
if(tx==x2&&ty==y2&&b.step<=k+1)
{
printf("yes\n");
return;
}
tx+=d[i][0];
ty+=d[i][1];
}
}
}
printf("no\n");
}

int main()
{
int T;
scanf("%d",&T);
while (T--)
{
scanf("%d%d",&n,&m);
init();
bfs();
}
return 0;
}