HDU1175 连连看(DFS)

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<algorithm>
#include <stdio.h>  
#include <string.h> 
int a[1001][1001],vis[1001][1001];
int n,m,x1,x2,y1,y2,flag;
using namespace std;
void dfs(int x,int y,int d,int k)//d(1 up 2 down 3 left 4 right)
{
  if(flag) return;
  if(k>=3) return;
  if(x<=0||y<=0||x>n||y>m) return;//此处x不是x轴！ 
  /* |-------->y
     |
     |
     |
     v
     x           读入顺序（愚昧的我）*/
  if(x==x2&&y==y2)//如果找到了 ，flag设为1 
  {
      flag=1;
      return;
  }
  if(k==2)//如果已经折了两次，则当前直线必定经过目标点（废话），若不则返回。 
  {
   if(!(d==1&&x>x2&&y==y2||d==2&&x>x2&&y==y2||d==3&&y>y2&&x==x2||d==4&&y<y2&&x==x2)) return; 
  }
  if(a[x][y]!=0) return;//如果非0为障碍，不能通过，返回
  if(vis[x][y]) return;//如果已经经过了，返回
  vis[x][y]=1;//标记为已经过
  if(d==1)//现在向上走 
  {
   dfs(x-1,y,1,k);
   dfs(x+1,y,2,k+1);//方向改变了，k+1 
   dfs(x,y-1,3,k+1);
   dfs(x,y+1,4,k+1);    
  } 
    if(d==2)//现在向下走 
  {
   dfs(x-1,y,1,k+1);
   dfs(x+1,y,2,k);
   dfs(x,y-1,3,k+1);
   dfs(x,y+1,4,k+1);    
  } 
    if(d==3)//现在向左走 
  {
   dfs(x-1,y,1,k+1);
   dfs(x+1,y,2,k+1); 
   dfs(x,y-1,3,k);
   dfs(x,y+1,4,k+1);    
  } 
    if(d==4)//现在向右走 
  {
   dfs(x-1,y,1,k+1);
   dfs(x+1,y,2,k+1);
   dfs(x,y-1,3,k+1);
   dfs(x,y+1,4,k);    
  } 
  vis[x][y]=0;//重新标记为未经过 
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        if(!n&&!m)//如果均为0 
        {
            break;
        }
        int i,j;
        for (i=1;i<=n;i++)
        {
            for(j=1;j<=m;j++)
            {
                scanf("%d",&a[i][j]);
            }
        }
        int t;
        scanf("%d",&t);
        while(t--)
        {
            flag=0;
            memset(vis,0,sizeof(vis));
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            if(x1==x2&&y1==y2&&a[x1][y1]!=0)//如果在同一坐标则是错的 
            {
               printf("NO
");  
            }
            else 
            {
            if(a[x1][y1]==a[x2][y2]&&a[x1][y1])//不同点，要满足值相同且不为0 
            {
                dfs(x1-1,y1,1,0);  
                dfs(x1+1,y1,2,0);  
                dfs(x1,y1-1,3,0);  
                dfs(x1,y1+1,4,0);  
                if(flag)  
                    printf("YES
");  
                else  
                    printf("NO
");  
            }
            else printf("NO
");  
            }
        }
    }
}