2021年度训练联盟热身训练赛第四场 Connect3

题目链接：训练联盟4 B

题目大意

有一个4*4的棋盘，可以看成他是竖着的，下棋的时候其实是从下到上堆上去的，也就是说要想下到(3,1)位置，前提是(2,1)和(1,1)已经有棋子填着，所以说显然，刚开始下棋，只能在第一行下，也就是说(1,1),(1,2),(1,3),(1,4)。如果说场上出现同一行或者同一列挥着对角线上相邻的三个棋子同色，则结束游戏，那中颜色的玩家获胜。然后给你一个x表示黑棋先下的位置是(1,x)，然后是a,b，表示白棋最后下的位置，问当白棋下最后一步赢了时，最后棋盘上棋子的所有可能出现的情况。

思路

直接暴搜，不过由于每次选择下一步走棋都是从头开始找的，所以说回溯的时候会出现重复的状态，所以将这个状态用hash值存一下，记map就完事了。直接暴力判相等，就是从这个位置开始往右走两格，往下走两格，往左下角走两个，往右下角走两个，这几个数是不是同色。

ac代码

#include<bits/stdc++.h>
using namespace std;
mt19937_64 rng(time(0));
#define io cin.tie(0);ios::sync_with_stdio(false);
#define ok(x, y) x >= 1 && x <= n && y >= 1 && y <= m
#define debug(x) cout<<#x<<"="<<x<<endl
#define lowbit(x) x&(-x)
#define pii pair<int,int>
#define mk make_pair
#define ll long long
#define ull unsigned long long
#define rs p<<1|1
#define ls p<<1
const int maxn = 1e6 + 5;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const ll INF = 1e18;
inline ll read(){
    ll p=0,f=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){p=(p<<1)+(p<<3)+(c^48),c=getchar();}
    return f*p;
}
void print(__int128 x){
    if(x<0) {putchar('-'); x=-x;}
    if (x>9) print(x/10);
    putchar('0'+x%10);
}

int vis[5][5];
int x, n, m, bas = 233;
ull hsh(){ //bas取3也可，因为每个位置一共就三种状态，3的16次也才4e7，都不会超int
    ull tmp = 0;
    for(int i = 1; i <= 4; i ++){
        for(int j = 1; j <= 4; j ++){
            tmp = tmp * bas + vis[i][j];
        }
    }
    return tmp;
}
unordered_map<ull, int> mp; //记状态
int okk(int s){ //暴力判相等，边界时4 4
    for(int i = 1; i <= 4; i ++){
        for(int j = 1; j <= 4; j ++){
            if(i + 2 <= 4 && vis[i][j] == s && vis[i + 1][j] == s && vis[i + 2][j] == s) return 1;
            if(j + 2 <= 4 && vis[i][j] == s && vis[i][j + 1] == s && vis[i][j + 2] == s) return 1;
            if(i + 2 <= 4 && j + 2 <= 4 && vis[i][j] == s && vis[i + 1][j + 1] == s && vis[i + 2][j + 2] == s) return 1;
            if(i - 2 >= 1 && j + 2 <= 4 && vis[i][j] == s && vis[i - 1][j + 1] == s && vis[i - 2][j + 2] == s) return 1;
        }
    }
    return 0;
}
int ans = 0;
void dfs(int s){
    ull st = hsh();
    if(mp.count(st)) return; //判记录是否存在过
    mp[st] ++;
    if(okk(2) || okk(1) || vis[n][m]){ //有三个相邻同色，不管是谁都要结束游戏，当n,m状态改变了也要结束游戏
        if(vis[n][m] == 2 && okk(2)) ans ++; //白棋获胜了
        return;
    }
    int i, j;
    for(j = 1; j <= 4; j ++){
        i = 1;
        while(i <= 4 && vis[i][j]) i++;
        if(i <= 4){ //该列上有空余位置课下棋则回溯
            vis[i][j] = s;
            dfs(3 - s); //1的下一个玩家是2，2的下一个玩家是1，那么就可以直接写成3-s
            vis[i][j] = 0;
        }
    }

}
void solve(){
    cin >> x >> n >> m;
    vis[1][x] = 1; //标记1为黑色，2为白色
    dfs(2); //下一个就是白色
    cout << ans << endl;
}

int main(){
    // freopen("1.in", "r", stdin);
    // freopen("std.out", "w", stdout);
    // cout << fixed << setprecision(6)
    io;
    solve();
    return 0;
}