hdoj1175__连连看

连连看

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

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
  
  

 
注意：起点或终点是0，输出NO，起点颜色不等于终点的输出NO。dfs中的剪枝，路径的段必须小于等于3，求路径的段时，记录上一个状态的方向，若相同，段数不变，若不同，段数加1.

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

int map[1010][1010];
bool visit[1010][1010];
int n, m, f;
int dx[] = {0, 1, 0, -1};
int dy[] = {1, 0, -1, 0};
int x_st, y_st, x_end, y_end;
int xiang;

void dfs(int x, int y, int sum)
{
	if (x == x_end && y == y_end)
	{
		if (sum <= 3)//段数小于等于3
			f = 1;
		return ;
	}
	if(f == 1)
		return ;
	if (sum > 3)//剪枝，段数大于2，直接返回
		return ;
	if(sum == 3 && x != x_end && y != y_end)//剪枝，段数到3，但是还没到终点，直接返回
		return ;
	for (int i = 0; i < 4; i++)
	{
		int k = x + dx[i];
		int l = y + dy[i];
		int t = xiang;//记录方向
		if(k >= 1 && l >= 1 && k <= n && l <= m && (map[k][l] == 0 || k == x_end && l == y_end) && !visit[k][l])
		{
			visit[k][l] = 1;
			if (xiang != i + 1)//判断方向
			{
				xiang = i + 1;//更新方向
				dfs(k, l, sum + 1);
			}
			else
			{
				xiang = i + 1;
				dfs(k, l, sum);
			}
			visit[k][l] = 0;
		}
		xiang = t;//方向还原
	}
}

int main()
{
	while (scanf("%d%d", &n, &m) == 2 && n != 0)
	{
		memset(map, 0, sizeof(map));
		for (int i = 1; i <= n; i++)
			for (int j = 1; j <= m; j++)
				scanf("%d", &map[i][j]);
		int cun;
		scanf("%d", &cun);
		while (cun--)
		{
			memset(visit, 0, sizeof(visit));
			f = 0;
			scanf("%d%d%d%d", &x_st, &y_st, &x_end, &y_end);
			if (map[x_st][y_st] == 0 || map[x_end][y_end] == 0)
			{
				printf("NO\n");
				continue;
			}
			if (map[x_st][y_st] != map[x_end][y_end])
			{
				printf("NO\n");
				continue;
			}
			visit[x_st][y_st] = 1;
			dfs(x_st, y_st, 0);
		    if (f == 1)
				printf("YES\n");
		    else 
			    printf("NO\n");
		}
	}
	return 0;
}