HDU 1175 连连看 (搜索 DFS)

连连看

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

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<string.h>
#include<cmath>
using namespace std;
#define up 1
#define down 2
#define right 3
#define left 4
int direct[5][2]={{0,0},{-1,0},{1,0},{0,1},{0,-1}};
int **board;
bool dfs(int x1,int y1,int x2,int y2,int direction ,int change)
{
    //剪枝
   if(change>3)
    return false;
    //可以消除
   if(x1==x2&&y1==y2)
      return true;
   for(int i=1;i<=4;i++)
   {
    //不走来时的路径
      if(direction==down&&i==up)
        continue;
      else if(direction==up&&i==down)
        continue;
      else if(direction==right&&i==left)
        continue;
      else if(direction==left&&i==right)
        continue;
      int xx,yy;
      xx = x1+direct[i][0];
      yy = y1+direct[i][1];
      //如果为0或者为目的地
      if((board[xx][yy]==0)||((board[xx][yy]==board[x2][y2]&&((abs(xx-x2)+abs(yy-y2))==0))))
      {
        if(direction!=i)
        {
          change++;
        }
        //深搜
        bool result = dfs(xx,yy,x2,y2,i,change);
        if(result)
             return true;
        else if(direction!=i)
         change--;
      }
   }
   return false;
}
int main()
{
   int row,col,testNum,x1,y1,x2,y2;
   while(cin>>row&&cin>>col&&row&&col)
   {
       board = new int*[row+1];
       for(int i=0;i<=row;i++)
        board[i] = new int[col];
        memset(board[0],-1,col*sizeof(board[0][0]));
       for(int i = 1;i<=row;i++)
        for(int j =1;j<=col;j++)
           cin>>board[i][j];
       cin>>testNum;
       for(int i = 1;i<=testNum;i++)
       {
           cin>>x1>>y1>>x2>>y2;
           if(board[x1][y1]!=board[x2][y2]||board[x1][y1]==0||board[x2][y2]==0)
           {
               cout<<"NO"<<endl;
               continue;
           }
           bool result = dfs(x1,y1,x2,y2,0,0);
           if(result)
            cout<<"YES"<<endl;
           else
            cout<<"NO"<<endl;
       }
        for(int i = 0;i<=row;i++)
           delete[] board[i];

   }
}