HDU-1175 连连看

连连看

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

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 41 2 3 40 0 0 04 3 2 141 1 3 41 1 2 41 1 3 32 1 2 43 40 1 4 30 2 4 10 0 0 021 1 2 41 3 2 30 0

 

Sample Output

YESNONONONOYES

题解：这题用的dfs做，但是有个难点就是判断转弯数，这里就用0，1，2，3来表示不同的方向，然后在递归的时候如果方向不同转弯数就加1。还有一点是循环要从1开始，因为这里的询问的数字是行、列，与数组下标不同。然后加一些剪枝判断返回就可以了。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;

const int maxn = 1010;
int pic[maxn][maxn];
int vis[maxn][maxn];
int x1,y1,x2,y2,n,m,turn;
int dix[4] = {0,1,0,-1};
int diy[4] = {1,0,-1,0};
int flag;

void dfs(int x,int y,int turn,int dir)
{
    if(flag)
        return;
    if(x < 1 || x > n || y < 1 || y > m || vis[x][y])
        return;
    if(turn == 2 && x != x2 && y != y2)
        return;
    if(turn > 2)
        return;
    if(turn <= 2 && x == x2 && y == y2)
    {
        printf("YES\n");
        flag = 1;
        return;
    }
    if(pic[x][y] != 0 && (x != x1 || y != y1))
    {
        return;
    }
    vis[x][y] = 1;
    for(int i = 0;i < 4;i++)
    {
        int dx = x + dix[i];
        int dy = y + diy[i];
        if(i == dir)
        {
            dfs(dx,dy,turn,i);
        }
        else
        {
            dfs(dx,dy,turn + 1,i);
        }
    }
    vis[x][y] = 0;
    return;
}

int main()
{
    while(scanf("%d%d",&n,&m) && n && m)
    {
        for(int i = 1;i <= n;i++)
        {
            for(int j = 1;j <= m;j++)
            {
                scanf("%d",&pic[i][j]);
            }
        }
        int k;
        scanf("%d",&k);
        while(k--)
        {
            flag = 0;
            memset(vis,0,sizeof(vis));
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            if(x1 == x2 && y1 == y2)
            {
                printf("NO\n");
                continue;
            }
            if(x1 < 1 || x1 > n || y1 < 1 || y1 > m || x2 < 1 || x2 > n || y2 < 1 || y2 > m)
            {
                printf("NO\n");
                continue;
            }
            if(pic[x1][y1] == pic[x2][y2] && pic[x1][y1])
            {
                for(int i = 0;i < 4;i++)
                {
                    dfs(x1,y1,0,i);
                }
                if(!flag)
                {
                    printf("NO\n");
                }
            }
            else
            {
                printf("NO\n");
            }
        }
    }
    return 0;
}