dfs（hdu1175 连连看）

连连看

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

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<cstring>
using namespace std;
int n,m;
int map[1005][1005];
bool vis[1005][1005];
struct node
{
	int x;
	int y;
};
node a,b;
int dx[4]={-1,1,0,0};
int dy[4]={0,0,1,-1};
bool ok(int x,int y)
{
	if(x>0&&x<=n&&y>0&&y<=m&&!vis[x][y]&&(map[x][y]==0||(map[x][y]!=0&&x==b.x&&y==b.y)))
	return true;
	else return false;
}
bool dfs(int x,int y,int dre,int num)//dre表示当前方向
{
	bool flag;
	vis[x][y]=true;
	if(num>2) return false;
	if(x==b.x&&y==b.y) return true;
	if(num==2&&x!=b.x&&y!=b.y) return false;
	for(int i=0;i<4;i++)
	{
		int xx=x+dx[i];
		int yy=y+dy[i];
		if(ok(xx,yy))
		{
			if(dre==-1) flag=dfs(xx,yy,i,0);
			else if(dre==i) flag=dfs(xx,yy,dre,num);
			else flag=dfs(xx,yy,i,num+1);
			vis[xx][yy]=false;
			if(flag) return true;
		}
	}
	return false;
}
int main()
{
	while(cin>>n>>m&&(n||m))
	{
		for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			cin>>map[i][j];
		}
		int k;
		cin>>k;
		for(int i=0;i<k;i++)
		{
			cin>>a.x>>a.y>>b.x>>b.y;
			if((a.x==b.x&&a.y==b.y)||map[a.x][a.y]==0||map[b.x][b.y]==0||map[a.x][a.y]!=map[b.x][b.y])
			{
				cout<<"NO"<<endl;
				continue;
			}
			memset(vis,0,sizeof(vis));
			bool flag=dfs(a.x,a.y,-1,0);
			if(flag) cout<<"YES"<<endl;
			else cout<<"NO"<<endl;
		}
	}
	return 0;
}