HDOJ 1175 连连看 (DFS)

连连看

Time Limit: 20000/10000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)

Total Submission(s): 359 Accepted Submission(s): 133

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

#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<queue>  
#include<algorithm>  
#include<cmath>  
using namespace std;

int map[1005][1005];
int n, m;
int a1, b1, a2, b2;
int v[1005][1005];
int dir[4][2] = { 0,1,0,-1,1,0,-1,0 };
bool flag;

void dfs(int x1, int y1,int t,int d)//t是转弯 d是方向
{
	if (flag)return;//剪枝
	if (x1>n || x1<1 || y1<1 || y1>m )return;
	if (t > 2)return;//转弯多了就退出
	if (t == 2 && x1 != a2&&y1 != b2)return;//没有这个条件就WA了
	v[x1][y1] = 1;
	for (int i = 0; i < 4; i++)//同方向优先 转弯次数少
	{
		int tx, ty;
		tx = x1 + dir[i][0];
		ty = y1 + dir[i][1];
		if (tx == a2&&ty == b2) { flag = true; return; }
		if (tx > 0 && ty > 0 && tx <= n&&ty <= m&&v[tx][ty] == 0 && map[tx][ty] == 0)
		{
			if (d == -1) {
				 dfs(tx, ty, t, i); 
			}
			if (d == i) {
				 dfs(tx, ty, t, i); 
			}
		}
	}
	for (int i = 0; i < 4; i++)//不是同方向的
	{
		int tx, ty;
		tx = x1 + dir[i][0];
		ty = y1 + dir[i][1];
		if (tx == a2&&ty == b2) { flag = true; return; }
		if (tx > 0 && ty > 0 && tx <= n&&ty <= m&&v[tx][ty] == 0 && map[tx][ty] == 0)
		{
			if (d != i&&d!=-1) {
				 dfs(tx, ty, t + 1, i); 
			}
		}
	}
	v[x1][y1] = 0;
	return;
}

int main()
{
	while (scanf("%d%d",&n,&m)!=EOF)
	{
		if (n == 0 && m == 0)break;
		for (int i = 1; i <= n; i++)//从1开始
		{
			for (int j = 1; j <= m; j++)
			{
				scanf("%d", &map[i][j]);
			}
		}
		int q;
		scanf("%d", &q);
		for (int i=0;i<q;i++)
		{
			memset(v, 0, sizeof(v));
			scanf("%d%d%d%d", &a1, &b1, &a2, &b2);
			v[a1][b1] = 1;
			if (map[a1][b1] != map[a2][b2] || map[a1][b1] == 0 ||(a1==a2&&b1==b2))printf("NO\n");//不相等 同一起点 为0 都是不行的
			else 
			{
				flag = false;
				dfs(a1, b1, 0, -1); //-1是初始态
				if (flag)printf("YES\n");
				else printf("NO\n");
			}

		}

	}

	return 0;
}