【DFS】连连看

连连看

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

Output

YES
NO
NO
NO
NO
YES

思路

• 可DFS可BFS，这里记录一种DFS方法
• 题意直截了当，别看是1000*1000的地图，其实时间给了10秒，所以普通DFS即可
• 关键是解决“拐弯”问题，最多只能拐两次，就是说方向最多变两次
• 那么在DFS函数中加上方向和拐弯次数的参数即可

代码如下

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

const int to[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};

vector<vector<int>> mp;
vector<vector<bool>> vis;
bool is_ok;
int n, m, k;

//DFS函数参数：起始位置坐标，终点位置坐标，拐弯次数，方向
void dfs(int x1, int y1, int x2, int y2, int d, int t) {
    if (d > 3 || is_ok) return;//由于第一次判断是否拐弯是一定要拐的，所以拐弯次数在比较时要加1
    if (d == 3 && x1 != x2 && y1 != y2) return;//剪枝，当拐弯次数恰为3但是还没到与目的位置的同一直线时，赶紧返回,少了这句话就是93ms与8892ms的差距。。。，不过还是可以过，没加也没关系
    if (x1 == x2 && y1 == y2) {
        is_ok = true;
        return;
    }
    for (int i = 0; i < 4; i++) {
        int tx = x1 + to[i][0];
        int ty = y1 + to[i][1];
        if (tx < 1 || tx > n || ty < 1 || ty > m) continue;
        if (vis[tx][ty] || (mp[tx][ty] && (tx != x2 || ty != y2))) continue;
        vis[tx][ty] = true;
        if (i == t) {
            dfs(tx, ty, x2, y2, d, i);
        } else {
        	//第一次DFS的时候，t=-1,i=0,是一定会拐弯一次的，所以拐弯次数判断标准加1，变为3
            dfs(tx, ty, x2, y2, d + 1, i);
        }
        vis[tx][ty] = false;
    }
}

void solve() {
    int i = 0, j = 0;
    int x1 = 0, y1 = 0, x2 = 0, y2 = 0;
    
    while (~scanf("%d%d", &n, &m) && n && m) {
        //使用vector动态开辟数组并输入，缺点是代码长了点
        mp.resize(n + 1);
        vis.resize(n + 1);
        for (i = 0; i <= n; i++) {
            vis[i].resize(m + 1, false);
            mp[i].resize(m + 1);
            if (i == 0) continue;
            for (j = 1; j <= m; j++) {
                scanf("%d", &mp[i][j]);
            }
        }
        
        scanf("%d", &k);
        while (k--) {
            scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
            
            //剪枝，明显不符合情况的就直接输出结果
            if ((mp[x1][y1] != mp[x2][y2]) ||
                (mp[x1][y1] == 0 && mp[x2][y2] == 0)) {
                printf("NO\n");
                continue;
            }
            
            //变量初始化之后就开始搜索
            is_ok = false;//是否找到合法的连线
            vis[x1][y1] = true;//这个貌似没用，但感觉写了更完整
            dfs(x1, y1, x2, y2, 0, -1);
            printf(is_ok ? "YES\n" : "NO\n");
            
            //清洗vis数组，为下一次搜索初始化
            for (i = 0; i <= n; i++) {
                fill(vis[i].begin(), vis[i].end(), false);
            }
        }
    }
}

int main(void) {
    solve();
    return 0;
}