连连看(搜索)

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

• 因为最多折2次，所以vis初始值为3，如果折数小于vis，就进行覆盖
• 因为可能覆盖的太多，会爆队列，用优先队列优化
• 提前剪枝，如果两点相同，有一点是0，两点值不同，3种情况直接NO

#include<queue>
#include<bits/stdc++.h>
using namespace std;
#define INF 0xfffffff;
int n, m, t;
int x1, y1, x2, y2;
int a[1005][1005];
int vis[1005][1005];
int dir[4][2] = {{1,0},{0,1},{-1,0},{0,-1}};
struct node{
	int x, y;
	int dir, cnt;
}nex, xia;
struct cmp{
	bool operator()(node &a, node &b){
		return a.cnt > b.cnt;
	}
};
int bfs(){
	priority_queue<node,vector<node>,cmp>p;
	p.push({x1,y1,-1,0});
	vis[x1][y1] = 0;
	while(!p.empty()){
		nex = p.top();
		p.pop();
		if(nex.x == x2 && nex.y == y2) return 1;
		for(int i = 0; i < 4; i ++){
			xia.x = nex.x + dir[i][0];
			xia.y = nex.y + dir[i][1];
			xia.cnt=nex.cnt;
			if(xia.x < 1 || xia.y < 1 || xia.x > n || xia.y > m)continue;
			if(a[xia.x][xia.y] == 0 || (xia.x == x2 && xia.y == y2)){
				if(nex.dir != -1){
					if(nex.dir != i){
						xia.cnt ++;
					}
				}
				xia.dir = i;
				if(xia.cnt < vis[xia.x][xia.y]){
					p.push(xia);
					vis[xia.x][xia.y] = xia.cnt;
//					cout << xia.x << " " << xia.y << " " << xia.cnt << endl;
				}
			}
			
		}
	}
	return 0;
}
int main() {
    ios::sync_with_stdio(false);
    while(cin >> n >> m){
    	if(n == 0 && m == 0)break;
    	for(int i = 1; i <= n; i ++)
    		for(int j = 1; j <= m; j ++){
    			cin >> a[i][j];
			}
    	cin >> t;
    	while(t --){
    		for(int i = 1; i <= n; i ++)
    			for(int j = 1; j <= m; j ++)
    			vis[i][j] = 3;
    		cin >> x1 >> y1 >> x2 >> y2;
    		if((x1 == x2 && y1 == y2) || (a[x1][y1] != a[x2][y2]) || (a[x1][y1] == 0 || a[x2][y2] == 0)){
    			cout << "NO" << endl;
    			continue;
			}
			if(bfs() == 1)cout << "YES" << endl;
			else cout << "NO" << endl;
		}
	}
    return 0;
}