hdu 1175 bfs+priority_queue

连连看

如上图所示如果采用传统bfs的话，如果按照逆时针方向从（1，1）-->（3，4）搜索，会优先选择走拐四次弯的路径导致ans错误；

Time Limit: 20000/10000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 34028    Accepted Submission(s):
8438

Problem Description

“连连看”相信很多人都玩过。没玩过也没关系，下面我给大家介绍一下游戏规则：在一个棋盘中，放了很多的棋子。如果某两个相同的棋子，可以通过一条线连起来（这条线不能经过其它棋子），而且线的转折次数不超过两次，那么这两个棋子就可以在棋盘上消去。不好意思，由于我以前没有玩过连连看，咨询了同学的意见，连线不能从外面绕过去的，但事实上这是错的。现在已经酿成大祸，就只能将错就错了，连线不能从外围绕过。
玩家鼠标先后点击两块棋子，试图将他们消去，然后游戏的后台判断这两个方格能不能消去。现在你的任务就是写这个后台程序。

 

Input

输入数据有多组。每组数据的第一行有两个正整数n,m(0<n<=1000,0<m<1000)，分别表示棋盘的行数与列数。在接下来的n行中，每行有m个非负整数描述棋盘的方格分布。0表示这个位置没有棋子，正整数表示棋子的类型。接下来的一行是一个正整数q(0<q<50)，表示下面有q次询问。在接下来的q行里，每行有四个正整数x1,y1,x2,y2,表示询问第x1行y1列的棋子与第x2行y2列的棋子能不能消去。n=0,m=0时，输入结束。
注意：询问之间无先后关系，都是针对当前状态的！

 

Output

每一组输入数据对应一行输出。如果能消去则输出"YES",不能则输出"NO"。

 

Sample Input

3 4

1 2 3 4

0 0 0 0

4 3 2 1

4

1 1 3 4

1 1 2 4

1 1 3 3

2 1 2 4

 

3 4
0 1 4 3
0 2 4 1
0 0 0 0
2
1 1 2 4
1 3 2 3
0 0

 

Sample Output

YES
NO
NO
NO
NO
YES

由于题目要求只能转弯二次，如果单纯的bfs的话由于搜索方向的不同可能导致最优结果被覆盖，则wa

所以考虑优先队列，设置比较算子<，转弯次数越小的优先级越大

#include<bits/stdc++.h>
using namespace std;
int e[1005][1005],n,m,X1,X2,Y1,Y2;
int fx[4][2]={-1,0,1,0,0,-1,0,1};
bool vis[1005][1005];
struct node
{
int x,y,num,pre;
bool operator<(const node&a)const                    //设置优先级<的算子，即优先级越小，num越大

{
return num>a.num;
}
};
bool bfs()
{
priority_queue<node> q;
node temp,tmp;
temp.x=X1,temp.y=Y1,temp.pre=-1,temp.num=0;
q.push(temp);
while(!q.empty()){
tmp=q.top();
q.pop();
for(int i=0;i<4;i++){
temp=tmp;
int dx=temp.x+fx[i][0];
int dy=temp.y+fx[i][1];
temp.x=dx,temp.y=dy;
if(tmp.pre==-1) temp.pre=i;
else{
temp.pre=i;
if(tmp.pre<=1&&i>1) temp.num++;                  //由上下变为左右或左右变为上下时，转弯次数+1
if(tmp.pre>1&&i<=1) temp.num++;
}
if(dx==X2&&dy==Y2&&temp.num<=2) return true;
if(dx<1||dy<1||dx>n||dy>m||vis[dx][dy]||temp.num>2||e[dx][dy]) continue;
vis[dx][dy]=1;
q.push(temp);
}
}
return false;
}
int main()
{
int i,j,flag,t;
while(cin>>n>>m&&n&&m){
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
scanf("%d",&e[i][j]);
scanf("%d",&t);
while(t--){memset(vis,0,sizeof(vis));
scanf("%d%d%d%d",&X1,&Y1,&X2,&Y2);
vis[X1][Y1]=1;
if(e[X1][Y1]!=e[X2][Y2]||e[X1][Y1]==0||e[X2][Y2]==0||(X1==X2&&Y1==Y2)) {puts("NO");continue;}
bfs()?puts("YES"):puts("NO");
}
}
return 0;
}