【HDU 1175】连连看（DFS）

Description

“连连看”相信很多人都玩过。没玩过也没关系，下面我给大家介绍一下游戏规则：在一个棋盘中，放了很多的棋子。如果某两个相同的棋子，可以通过一条线连起来（这条线不能经过其它棋子），而且线的转折次数不超过两次，那么这两个棋子就可以在棋盘上消去。不好意思，由于我以前没有玩过连连看，咨询了同学的意见，连线不能从外面绕过去的，但事实上这是错的。现在已经酿成大祸，就只能将错就错了，连线不能从外围绕过。
玩家鼠标先后点击两块棋子，试图将他们消去，然后游戏的后台判断这两个方格能不能消去。现在你的任务就是写这个后台程序。

Input

输入数据有多组。每组数据的第一行有两个正整数n,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

题目大意

中文题

思路

卡了一下午，在判断转折次数上一直出错，最后无奈去看了题解，重新对着题解敲了一遍。其实判断转折次数只要判断下一次行走的方向和上一次行走的方向是否相同即可，因为我们定义了dir[x][y]数组用于表示前进的方向所以只需要判断x的值和行进前的x值是否相等就可以判断有没有转折。
另外一个难点就是剪枝，很容易发现如果转折次数为2次的时候如果目标点和当前所在的点不在同一个直线上，那么必定还要转折一次，这种情况可以直接返回。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn=1005;
int map[maxn][maxn],n,m,x1,y1,x2,y2,flag,q;
int vis[maxn][maxn];

void dfs(int x,int y,int z,int cnt)//z表示前进的方向，1:up,2:down,3:left,4:right 
{
    if(flag) return;
    if(cnt>2) return;
    if(x<=0||x>n||y<=0||y>m) return;
    if(x==x2&&y==y2)
    {
        flag=1;
        return;
    }
    if(cnt==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(map[x][y]!=0) return;
    if(vis[x][y]) return;
    vis[x][y]=1;
    if(z == 1)//确定一个前进方向，对4个方向分别dfs 
    {
        dfs(x-1,y,1,cnt);
        dfs(x+1,y,2,cnt+1);
        dfs(x,y-1,3,cnt+1);
        dfs(x,y+1,4,cnt+1);
    }
    else if(z == 2)
    {
        dfs(x-1,y,1,cnt+1);
        dfs(x+1,y,2,cnt);
        dfs(x,y-1,3,cnt+1);
        dfs(x,y+1,4,cnt+1);
    }
    else if(z == 3)
    {
        dfs(x-1,y,1,cnt+1);
        dfs(x+1,y,2,cnt+1);
        dfs(x,y-1,3,cnt);
        dfs(x,y+1,4,cnt+1);
    }
    else if(z == 4)
    {
        dfs(x-1,y,1,cnt+1);
        dfs(x+1,y,2,cnt+1);
        dfs(x,y-1,3,cnt+1);
        dfs(x,y+1,4,cnt);
    }
    vis[x][y] = 0;
}

int main()
{
    while(~scanf("%d %d",&n,&m))
    {
        if(!n&&!m) break;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                scanf("%d",&map[i][j]);
            }
        }
        scanf("%d",&q);
        for(int i=1;i<=q;i++)
        {
            flag=0;
            memset(vis,0,sizeof(vis));
            scanf("%d %d %d %d",&x1,&y1,&x2,&y2);
            if(x1==x2&&y1==y2&&map[x1][y1]!=0)
                printf("NO\n");
            else if(map[x1][y1]==map[x2][y2]&&map[x1][y1])
            {
                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;
}