HDU 1175 连连看（DFS+两种剪枝）

题目链接

情人节 干嘛？当然是刷题了！

连连看

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

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，当然我看评论区也有人用BFS。
虽然简单，但是如果直接套模板，毫无疑问是会超时10000ms。
由此可见，DFS剪枝的重要性
这里尝试了两种剪枝方法：
1.弱鸡本鸡的方法：没走过一个点，便更新走到节点的最小转弯次数：如果更小就更新，如果大于等于就剪枝。（327ms）

if(top.step>=mstep[top.x][top.y])
        return false;
    else{
        mstep[top.x][top.y]=top.step;
    }

2.大佬的方法：当一个点的转向次数为2时，判断此节点是否和终点在同一条直线上。（280ms）

if(top.step==2&&top.x!=D.x&&top.y!=D.y)
        return false;

完整代码：

#include <iostream>

#define MAXN 1005
#define MAXM 1005
using namespace std;
int n,m;
int maze[MAXN][MAXM];
bool vis[MAXN][MAXM];
int mstep[MAXN][MAXM];
int opt;

int X[]={0,0,1,-1};
int Y[]={1,-1,0,0};

struct node{
    int x,y;
    int way;
    int step;
}S,D;

bool check(int x,int y){
    if(x<1||x>n||y<1||y>m)
        return false;
    if(vis[x][y])
        return false;
    if(maze[x][y]!=0&&maze[x][y]!=maze[S.x][S.y])
        return false;
    return true;
}

bool dfs(node top){
    if(top.step>2){
        return false;
    }
    //大佬的剪枝
    /*if(top.step==2&&top.x!=D.x&&top.y!=D.y)
        return false;*/
    if(top.step<=2&&top.x==D.x&&top.y==D.y)
        return true;

    //我的剪枝：记录下此节点最小转向次数
    if(top.step>=mstep[top.x][top.y])
        return false;
    else{
        mstep[top.x][top.y]=top.step;
    }

    for(int i=0;i<4;i++){
        node tmp;
        tmp.x=top.x+X[i];
        tmp.y=top.y+Y[i];
        tmp.way=i;//记录方向
        tmp.step=top.step;
        if(tmp.way!=top.way)//方向不同，次数加1
            tmp.step=top.step+1;
        if(!check(tmp.x,tmp.y)) continue;
        //如果遇到相同数字的点，绕过去
        if(maze[tmp.x][tmp.y]==maze[S.x][S.y]&&(tmp.x!=D.x||tmp.y!=D.y)) continue;
        vis[tmp.x][tmp.y]=true;
        if(dfs(tmp))
            return true;//找到了，一直返回true，知道完全退出函数
        //没找到，回溯
        vis[tmp.x][tmp.y]=false;
    }
    return false;
}



int main()
{
    while(cin>>n>>m){
        if(!n&&!m) break;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                cin>>maze[i][j];
            }
        }
        cin>>opt;
        while(opt--){
            for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                mstep[i][j]=9999;
            memset(vis,false,sizeof(vis));
            cin>>S.x>>S.y>>D.x>>D.y;
            S.step=-1;
            S.way=-1;
            //简单判断
            if(maze[S.x][S.y]!=maze[D.x][D.y]||maze[S.x][S.y]==0||maze[D.x][D.y]==0||(S.x==D.x&&S.y==D.y)){
                cout<<"NO"<<endl;
                continue;
            }
            if(dfs(S))
                cout<<"YES"<<endl;
            else
                cout<<"NO"<<endl;
        }
    }
    return 0;
}