BFS 搜索 Problem 1009 连连看

Problem ID：1009 连连看

简单题意：一个棋盘上放了若干个棋子，如果两个相同（位置不同）的棋子能用一条线连起来，且转折次数不超过2次，则可将其消去。给出棋盘和棋子，以及试图消去的两个棋子位置。如果能消去，输出“YES”，如果不能，输出“NO”。

解题思路形成过程：利用BFS进行搜索，符合要求的下一步共有3个要求：

①：连线必须在棋盘内，不能从外面绕出去；

②：下一个目标区域的没有以任何方向走过，或者新的走法转折次数更少；

③：此区域无棋子（如果找到了目标直接return返回）。

注意：如果选择的两个棋子类型不同，或其中任意一个位置没有棋子，则须直接输出“NO”。

感想： 一个不是太难的题，WA了16次………… 改了N久…… 最后才发现WA的原因是如果是该输出“NO”，主函数里的条件该写成对应的-1 而不是0…… 反复检查居然没有发现…………

还有就是中间有一个情况没有考虑到…

一开始用的DFS，会超时，后来又改成BFS写的。不过如果剪枝处理的比较好，DFS应该也可以。

第一次见识和学习了三维数组，挺好使。

注意条件的整合，以提高代码效率。

无论如何…… 终于是AC了…

代码：

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
#define INF 10000
int cmap[1001][1001];
int mark[1001][1001][4];
int query[4];
int dir[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
int n,m;
int BFS()
{
queue<int>s;
s.push(query[0]);       //队列中4个元素为一组，分别是当前行(r)，当前列(c)，当前方向(state)，当前转折次数(cnt)。
s.push(query[1]);
s.push(-1);
s.push(-1);
while(!s.empty())
{
int r=s.front();
s.pop();
int c=s.front();
s.pop();
int state=s.front();
s.pop();
int cnt=s.front();
s.pop();

for(int i=0;i<4;++i){   //i也代表了移动的方向。i=0,1,2,3分别代表了上，下，左，右。
int t_cnt=cnt;
if(i!=state)
++t_cnt;
if(t_cnt>2)
continue;
if(r+dir[i][0]==query[2]&&c+dir[i][1]==query[3])
return 1;
if(r+dir[i][0]>=0&&r+dir[i][0]<n&&c+dir[i][1]>=0&&c+dir[i][1]<m)
if(mark[r+dir[i][0]][c+dir[i][1]][i]>t_cnt)       //没走过或者新的路径转折次数更少
if(cmap[r+dir[i][0]][c+dir[i][1]]==0){
// cout<<r<<' '<<c<<' '<<"go"<<endl;
s.push(r+dir[i][0]);
s.push(c+dir[i][1]);
s.push(i);
s.push(t_cnt);
mark[r+dir[i][0]][c+dir[i][1]][i]=t_cnt;
}
}
}
return -1;
}
int main()
{
freopen("1.txt","r",stdin);
while(1)
{
scanf("%d%d",&n,&m);
if(n==0)
return 0;
for(int i=0;i<n;++i)
for(int j=0;j<m;++j)
cin>>cmap[i][j];
int q;
scanf("%d",&q);
for(int i=0;i<q;++i){
scanf("%d%d%d%d",&query[0],&query[1],&query[2],&query[3]);
--query[0];
--query[1];
--query[2];
--query[3];
memset(mark,INF,sizeof(mark));//INF
if(cmap[query[0]][query[1]]!=cmap[query[2]][query[3]]||cmap[query[0]][query[1]]==0||cmap[query[2]][query[3]]==0||(query[0]==query[2]&&query[1]==query[3]))
printf("NO\n");
else if(BFS()==1)
printf("YES\n");
else if(BFS()==-1)
printf("NO\n");
}
}
}