hdu 1175 连连看（dfs+剪枝）

连连看

  HDU - 1175 

“连连看”相信很多人都玩过。没玩过也没关系，下面我给大家介绍一下游戏规则：在一个棋盘中，放了很多的棋子。如果某两个相同的棋子，可以通过一条线连起来（这条线不能经过其它棋子），而且线的转折次数不超过两次，那么这两个棋子就可以在棋盘上消去。不好意思，由于我以前没有玩过连连看，咨询了同学的意见，连线不能从外面绕过去的，但事实上这是错的。现在已经酿成大祸，就只能将错就错了，连线不能从外围绕过。 
玩家鼠标先后点击两块棋子，试图将他们消去，然后游戏的后台判断这两个方格能不能消去。现在你的任务就是写这个后台程序。 

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 <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

int map[1005][1005];
int vis[1005][1005];

int dx[]={1,-1,0,0};
int dy[]={0,0,1,-1};

int n,m,flag;
int sx,sy,ex,ey;
//                       朝向     转弯次数 
void dfs(int i,int j,int face,int times)
{
	//printf("%d %d %d %d %d  ----\n",i,j,map[i][j],face,times);
	
	if(i<0 || i>=n || j<0 || j>=m) //判断边界
		return ;
	if(vis[i][j] || times>2 || flag)//判断是否走过，转弯次数，是否已经有满足条件的情况 
		return ;
	if(times==2 && (i-ex)!=0 && (j-ey)!=0)//这个剪枝很重要啊 ，不然会超时 
		return;
	if(i==ex && j==ey)//存在满足条件的情况 
	{
		flag=1;
		return ;
	}
	
	/* 
	if(map[i][j]!=0 && (i!=sx || j!=sy)) // 是 (i!=sx || j!=sy) || ,不是 && !!!
        return ;                      	 //判断这个点能不能走,可以代替下面的if语句 
	*/
	
	vis[i][j]=1;
	for(int k=0;k<4;k++)
	{
		int xx,yy;
		xx=i+dx[k];
		yy=j+dy[k];
		if(map[xx][yy]==0 || (xx==ex && yy==ey))//判断点(xx,yy)是否可走 
		{
			if(face!=-1 && face!=k) //判断是否转弯 
				dfs(xx,yy,k,times+1);
			else
				dfs(xx,yy,k,times);	
		}
	}
	vis[i][j]=0;	
}

int main()
{
    while(scanf("%d %d",&n,&m) && (n+m))
    {	
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                scanf("%d",&map[i][j]);
        int k;
        scanf("%d",&k);
        while(k--)
        {
			memset(vis,0,sizeof(vis));
            scanf("%d %d %d %d",&sx,&sy,&ex,&ey);
            sx--,sy--,ex--,ey--;
            if(map[sx][sy]!=map[ex][ey] || !map[sx][sy] || !map[ex][ey])
            {
                printf("NO\n");
                continue;
            }
            flag=0;
			dfs(sx,sy,-1,0);
            if(flag==1)
                printf("YES\n");
            else
                printf("NO\n");
        }    
    }
    return 0;
}