From:http://acm.hdu.edu.cn/showproblem.php?pid=1175
题意:0为空,数字1~n代表不同的棋子,给出一个棋盘和q对两个棋子的坐标(x1,y1)、(x2,y2),问能否消除。
思路:BFS向一个方向搜,若搜不到则转向,若能在转向次数<=2时搜到,则YES,反之NO。
#include <bits/stdc++.h>
using namespace std;
int n,m;
int maze[1050][1050];
int dr[4]={1,0,-1,0};
int dc[4]={0,1,0,-1};
int vis[1050][1050];
int ok;
struct node{
int x,y,cnt;
}st,ed;
void bfs()
{
queue <node> q;
st.cnt=-1;//这种写法会导致转向次数初始为1,需-1
q.push(st);
node tmp;
while(!q.empty())
{
tmp=q.front();
vis[tmp.x][tmp.y]=1;
q.pop();
for(int i=0;i<4;i++)
{
node nxt;
nxt.x=tmp.x+dr[i];
nxt.y=tmp.y+dc[i];
//下面这个while与之前做的多向同找不同,需先往一个方向找至尽头,若找不到,则转向+1并转向
while(nxt.x>=0&&nxt.x<n&&nxt.y>=0&&nxt.y<m&&(maze[nxt.x][nxt.y]==0||(nxt.x==ed.x&&nxt.y==ed.y)))
{
if(!vis[nxt.x][nxt.y]){
nxt.cnt=tmp.cnt+1;//此tmp.cnt为上面定义的上次的cnt,比如上次的cnt是0,此次就是1,然后入队,直到再用此队列元素时,这个元素是已经转过一次向的
vis[nxt.x][nxt.y]=1;
q.push(nxt);
if(nxt.x==ed.x&&nxt.y==ed.y&&nxt.cnt<=2)
{
ok=1;
return ;
}
}
nxt.x+=dr[i],nxt.y+=dc[i];//在此方向上往深处找
}
}
}
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF&&n+m){
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
{
cin>>maze[i][j];
}
}
int q;
scanf("%d",&q);
while(q--!=0)
{
scanf("%d%d%d%d",&st.x,&st.y,&ed.x,&ed.y);
ok=0;
memset(vis,0,sizeof(vis));
st.x--,st.y--,ed.x--,ed.y--;//此处是棋盘按0~n-1读入的
if(maze[st.x][st.y]!=maze[ed.x][ed.y])//剪枝
{
printf("NO\n");
continue;
}
if(maze[st.x][st.y]==0||maze[ed.x][ed.y]==0||(st.x==ed.x&&st.y==ed.y))//剪枝
{
printf("NO\n");
continue;
}
// printf("x1:%d y1:%d x2:%d y2:%d\n",x1,y1,x2,y2);
bfs();
if(ok) printf("YES\n");
else printf("NO\n");
}
}
return 0;
}