hdu 1175 连连看dfs(记录转弯次数)

点击打开链接

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int M =1100;
typedef struct{
int x;
int y;
}Move;
Move movE[4]={{-1,0},{1,0},{0,-1},{0,1}};
int g[M][M],n,m,flag,visit[M][M];
//last=0~3 up down left right
void dfs(int a,int b,int c,int d,int turn,int last)// turn 记录转弯次数 last记录上次方向
{
//cout<<a<<"，"<<b<<"@"<<turn<<endl;
if(flag) return;
if(a==c&&b==d)
{
flag=1;
return;
}
if(last==-1)
{
for(int i=0;i<4;i++)
{
int x=a+movE[i].x;
int y=b+movE[i].y;
if(x>=1&&x<=n&&y>=1&&y<=m&&!visit[x][y]&&(x==c&&y==d||g[x][y]==0))
{

visit[x][y]=1;
dfs(x,y,c,d,0,i);
visit[x][y]=0;//
}

}
}
else
{
for(int i=0;i<4;i++)
{
if(turn<=2)
{

int x=a+movE[i].x;
int y=b+movE[i].y;
if(x>=1&&x<=n&&y>=1&&y<=m&&!visit[x][y]&&(x==c&&y==d||g[x][y]==0)) // g[x][y]为0 或者是终点
{
if(i==last)
{
visit[x][y]=1;
dfs(x,y,c,d,turn,i);
visit[x][y]=0;//该方向不行 则回溯
}
else
{
if(turn+1<=2)
{
visit[x][y]=1;
dfs(x,y,c,d,turn+1,i); //方向与上次不同 turn+1;
visit[x][y]=0;
}
}
}
}
}

}
}
int main()
{
while(cin>>n>>m&&(n+m))
{
memset(visit,0,sizeof(visit));
flag=0;//
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
cin>>g[i][j];
}
}
int q;
cin>>q;
while(q--)
{
flag=0;//
int a,b,c,d;
cin>>a>>b>>c>>d;
if(g[a][b]!=g[c][d]||g[a][b]==0||g[c][d]==0)
{
cout<<"NO"<<endl;
\
continue;
}
else
{
dfs(a,b,c,d,0,-1);
if(flag)
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}

}

}
return 0;
}