HDU 1175 连连看（搜索）

连连看

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

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增加一个变量记录上一个点的方向，用sum记录一共改了几次。。注意这里要加一个优化，否则会超时，其实也不能说优化。。。sum > 2的时候直接ruturn掉，sum == 2的时候如果x或者y不跟终点在一条线上说明还要变方向，肯定不会也return。。并且我是在dfs函数里输出yes的。。。所以只要找到一个输出yes之后，之后再找到就不能输出了，所以加一个if（flag） return。。如果不在dfs输出可以不加（最好加上，减少复杂度）

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
int dir[4][2] = {1,0,-1,0,0,1,0,-1};
const int maxn = 1e3 + 5;
int maze[maxn][maxn], n, m, q, book[maxn][maxn];
int sx, sy, ex, ey, flag;
void dfs(int x, int y, int pre, int sum)
{
    if(flag)  return ;  //这里是判断是否复合情况的，如果复合了就可以不用再搜下去了，加一个这个
    if(x == ex && y == ey && sum <= 2 )
       {
           // cout << maze[x][y] << endl << maze[ex][ey] << endl;
            printf("YES\n");
            flag = 1;
            return ;
        }
    if(sum >= 2 && ex != x && ey != y) return;
    for(int i = 0; i < 4; i++)
    {
        int tx = x + dir[i][0];
        int ty = y + dir[i][1];
        if(tx > 0 && ty > 0 && tx <= n && ty <= m && (maze[tx][ty] == 0 || tx == ex && ty == ey)&& !book[tx][ty]) //在路途中都要是0，最后一个可以不是0，可以这样写，这里也被卡过。。
        {
            book[tx][ty] = 1;
            if(pre == -1 || pre == i)  //-1说明最开始走
            {
                dfs(tx, ty, i, sum);
            }
            else if(pre != i)
            {
                dfs(tx, ty, i, sum+1);
            }
            book[tx][ty] = 0;
        }
    }
}
int main()
{
    while(~scanf("%d%d", &n, &m) , m+n)
    {
        for(int i = 1; i <= n; i++)
            for(int j = 1; j <= m; j++)
            scanf("%d", &maze[i][j]);
        scanf("%d", &q);
        while(q--)
        {
            scanf("%d%d%d%d", &sx, &sy, &ex, &ey);
            flag = 0;
            if(maze[sx][sy] == maze[ex][ey] && maze[sx][sy] != 0)  //这里就是一开始没想好的地方。。
            {
                dfs(sx,sy, -1, 0);
                if(!flag)
                    printf("NO\n");
            }
            else
                printf("NO\n");


        }
    }
    return 0;
}