hud 1175 连连看（dfs）

连连看

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

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
  

    

        连连看大家都应该玩过吧，如果写一个连连看游戏，就要用到搜索的算法，下面就是这题的解决方法。注意：记得剪枝，不然会超时，你玩连连看游戏不会出现按两个东西要等很久才能消吧，所以剪枝很重要！

链接： http://acm.hdu.edu.cn/showproblem.php?pid=1175

代码：

#include <iostream>
#include <stdio.h>
#include <memory.h>
using namespace std;

int n, m, t, k, z;
int x1, y1, x2, y2;
int a[1005][1005];
bool map[1005][1005], flag;

void init()     //初始化函数
{
    flag = false;
    memset(map, false, sizeof(map));
}

void dfs(int x, int y, int z, int k)    //坐标(x,y)
{
    if(flag) return;
    if(x<=0 || y<=0 || x>n || y>m) return;  //若超界，返回
    if(k>=3) return;        //若转弯数已经超过3次，返回
    if(x == x2 && y == y2)
    {
        flag = true;
        return;
    }
    if(k == 2)  //超强剪枝：若已经转两次弯，则目标坐标必须要在前进方向的前面，否则直接返回
    {
        if( !(z==1 && x>x2 && y==y2  ||  z==2 && x<x2 && y==y2
              ||  z==3 && y>y2 && x==x2  ||  z==4 && y<y2 && x==x2) )
            return;
    }
    if(a[x][y] != 0) return;    //如果该点不是0，则不能走，返回
    if(map[x][y]) return;       //如果该点已经走过，返回
    map[x][y] = true;       //标记该点为走过
    if(z == 1)  //上
    {
        dfs(x-1, y, 1, k);
        dfs(x+1, y, 2, k+1);
        dfs(x, y-1, 3, k+1);
        dfs(x, y+1, 4, k+1);
    }
    else if(z == 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);
    }
    else if(z == 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);
    }
    else if(z == 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);
    }
    map[x][y] = false;  //若深搜不成功，标记该点为未走过
}

int main()
{
    int i, j;
    while(scanf("%d %d", &n, &m))
    {
        if(!n && !m) break;
        for(i = 1; i <= n; i++)
            for(j = 1 ; j <= m; j++)
                scanf("%d", &a[i][j]);
        scanf("%d", &t);
        for(i = 0; i < t; i++)
        {
            init();
            scanf("%d %d %d %d", &x1, &y1, &x2, &y2);
            if(x1 == x2 && y1 == y2 && a[x1][y1] != 0) //若两点坐标是相同的，输出NO
                printf("NO\n");
            else if(a[x1][y1] == a[x2][y2] && a[x1][y1] != 0) //若两点的值相同且不为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\n");
                else printf("NO\n");
            }
            else printf("NO\n");
        }
    }

    return 0;
}