HDU 1175 连连看 DFS

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

Code:

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <string>
#include <cstring>
#include <map>
#include <cmath>
#include <vector>
#include <set>
#include <queue>
#define INF 0x3f3f3f3f
using namespace std;
static const int MAX_N = 5e4 + 5;
static const int dir[4][2] = { {0, 1}, {0, -1}, {1, 0}, {-1, 0} };
int n, m;
int maps[1005][1005];
bool vis[1005][1005];
struct Node{
	int x;
	int y;
	int flag;
	int dire;
	Node(){}
	Node(int _x, int _y, int _dire, int _flag) { x = _x; y = _y; dire = _dire; flag = _flag; }
};
bool inbound(int x, int y) {
	return (x >= 0 && x < n && y >= 0 && y < m);
}
bool bfs(int x1, int y1, int x2, int y2) {
	memset(vis, false, sizeof(vis));
	Node p1(x1, y1, 0, 0);
	vis[x1][y1] = true;
	queue<Node>q;
	q.push(p1);
	while (!q.empty()) {
		Node p2 = q.front();
		q.pop();
		if (p2.x == x2 && p2.y == y2) return true;
		for (int i = 0; i < 4; ++i) {
			p1.x = p2.x + dir[i][0];
			p1.y = p2.y + dir[i][1];
			if (dir[i][0]) p1.flag = -1;
			else p1.flag = 1;
			if (p1.flag + p2.flag) p1.dire = p2.dire;
			else p1.dire = p2.dire + 1;
			if (inbound(p1.x, p1.y) && !vis[p1.x][p1.y] && p1.dire <= 2){
				if (maps[p1.x][p1.y] == maps[x2][y2] && p1.x == x2 && p1.y == y2)
					return true;
				else if (maps[p1.x][p1.y] == 0) {
					q.push(p1);
					vis[p1.x][p1.y] = true;
				}
			}
		}
	}
	return false;
}
int main() {
	/*freopen("input.txt", "r", stdin);
	freopen("output.txt", "w", stdout);*/
	while (scanf("%d%d", &n, &m) != EOF) {
		if (n == 0 && m == 0) break;
		for (int i = 0; i < n; ++i) {
			for (int j = 0; j < m; ++j) scanf("%d", &maps[i][j]);
		}
		int q;
		scanf("%d", &q);
		while (q--) {
			int x1, y1, x2, y2;
			scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
			if (!inbound(x1 - 1, y1 - 1) || !inbound(x2 - 1, y2 - 1)) printf("NO\n");
			else if (maps[x1 - 1][y1 - 1] != maps[x2 - 1][y2 - 1] || maps[x1 - 1][y1 - 1] == 0 || maps[x2 - 1][y2 - 1] == 0) printf("NO\n");
			else if (bfs(x1 - 1, y1 - 1, x2 - 1, y2 - 1)) printf("YES\n");
			else printf("NO\n");
		}
	}
	return 0;
}