HDU-1175 连连看

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

思路：基本的dfs，注意剪枝，要不然会超时。

代码：

#include <stdio.h>
#include <string.h>
int e[1010][1010];
int book[1010][1010];
int flag;
int d1[8]={-1,0,1,0};
int d2[8]={0,1,0,-1};
int n,m,x2,y2;

void dfs(int x,int y,int d,int t)
{
	int tx,ty,i;
	if(t>2||flag==1)
	
		return ;
	if(t==2&&(x-x2)!=0&&(y-y2)!=0)//剪枝，判断转弯两次之后与目标点在不在一条直线上 
		return ;
	if(x==x2&&y==y2&&t<=2)
	{
		flag=1;
		return ;
	}
	for(i=0;i<4;i++)
	{
		tx=x+d1[i];
		ty=y+d2[i];
		if(tx<1||ty<1||tx>n||ty>m)
			continue;
		if(e[tx][ty]==0||(tx==x2&&ty==y2))
		{
			book[tx][ty]=1;
			if(d==-1||d==i)
				dfs(tx,ty,i,t);
			else
				dfs(tx,ty,i,t+1);
			book[tx][ty]=0;
		}  
	}	
}

int main()
{
	int x1,y1;
	int q,t,i,j;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		if(n==0&&m==0)
			break;
		memset(e,0,sizeof(e));
		for(i=1;i<=n;i++)
			for(j=1;j<=m;j++)
				scanf("%d",&e[i][j]);
		scanf("%d",&q);
		for(i=1;i<=q;i++)
		{
			flag=0;
			scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
			memset(book,0,sizeof(book));
			if(e[x1][y1]==e[x2][y2]&&e[x1][y1]!=0)
				dfs(x1,y1,-1,0);
			if(flag==1)
				printf("YES\n");
			else
				printf("NO\n");		
		}
	}
	return 0;
}