HDU 1175 连连看（dfs+剪枝）

连连看

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

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

Sample Output

YES
NO
NO
NO
NO
YES

分析

dfs，记录当前的方向和转弯次数，超出限制就return

剪枝：当次数为2次时，若当前点与目标不在同一直线则return

需要注意的是，需要判断出发点与目标点数值是否相同，而且不为0

#include<bits/stdc++.h>

using namespace std;

int ff[2][4]={0,0,1,-1,1,-1,0,0},x1,x2,y1,y2,flag,vis[1005][1005],mp[1005][1005],n,m,t;

void dfs(int x,int y,int f,int step)
{
    if(step>2||flag||(step==2&&(x-x2!=0)&&(y-y2!=0)))
    return;
    if(x==x2&&y==y2&&step<=2)
    {
        flag=1;
        return;
    }
    for(int i=0;i<4;i++)
    {
        int xx=x+ff[0][i];
        int yy=y+ff[1][i];
        if(xx>=0&&xx<n&&yy>=0&&yy<m&&(mp[xx][yy]==0||(xx==x2&&yy==y2))&&vis[xx][yy]==0)
        {
            vis[xx][yy]=1;
            if(i==f)
            {
                dfs(xx,yy,f,step);
            }
            else
            {
                dfs(xx,yy,i,step+1);
            }
            vis[xx][yy]=0;
        }
    }
}

int main()
{
    while(cin>>n>>m)
    {
        if(n==0&&m==0)
        break;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                cin>>mp[i][j];
            }
        }
        cin>>t;
        while(t--)
        {
            flag=0;
            cin>>x1>>y1>>x2>>y2;
            x1--;
            y1--;
            x2--;
            y2--;
            if((mp[x1][y1]!=mp[x2][y2])||mp[x1][y1]==0)
            {
                cout<<"NO"<<endl;
                continue;
            }
            vis[x1][y1]=1;
            for(int i=0;i<4;i++)
            {
                dfs(x1,y1,i,0);
            }
            vis[x1][y1]=0;
            if(flag)
            cout<<"YES"<<endl;
            else
            cout<<"NO"<<endl;
        }
    }
}