0
点赞
收藏
分享

微信扫一扫

蓝桥杯 基础训练-2n皇后问题

color_小浣熊 2022-01-09 阅读 43

解题思路:
蓝桥杯不错的一道暴力题,直接两个dfs暴搜就行,每次放一行,枚举列,如果先放白色的,那么放置完记得把地图标为不可放置,这样再跑黑色皇后的时候,就不会出现重复的放置。
talk is cheap, show me the code
在这里插入图片描述

//a ber
#include <bits/stdc++.h>
using namespace std;
const int N = 1e2 + 10;
int n, ok, mp[N][N];
int v1[N], v1x[N], v1y[N]; //黑色
int v2[N], v2x[N], v2y[N]; //白色
bool check1(int x, int i) {
    //同时满足 当前列,正对角线 反对角线 地图为1  才可放置
    return !v1[i] && !v1x[x + i] && !v1y[x - i + n] && mp[x][i];
}
bool check2(int x, int i) {
    return !v2[i] && !v2x[x + i] && !v2y[x - i + n] && mp[x][i];
}
void dfs2(int x) {
    //成功走到n就ok++ 
    if(x == n) {
        ok++;
        return;
    }
    for(int i = 0; i < n; i++) {
        if(check2(x, i)) {
            v2[i] = v2x[x + i] = v2y[x - i + n]  = 1;
            mp[x][i] = 0;
            dfs2(x + 1);
            v2[i] = v2x[x + i] = v2y[x - i + n]  = 0;
            mp[x][i] = 1;
        }
    }
}
void dfs1(int x) {
    if(x == n) {
        //黑色放置完成 进入白色搜索
        dfs2(0);
        return;
    }
    for(int i = 0; i < n; i++) {
        if(check1(x, i)) {
            v1[i] = v1x[x + i] = v1y[x - i + n]  = 1;
            mp[x][i] = 0; //记得把地图标记为不可走
            dfs1(x + 1);
            v1[i] = v1x[x + i] = v1y[x - i + n]  = 0;
            mp[x][i] = 1;
        }
    }
}
signed main() {
    scanf("%d", &n);
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++) {
            scanf("%d", &mp[i][j]);
        }
    }
    //进入第一次搜索 放置黑色 
    dfs1(0);
    printf("%d",ok);

    return 0;
}

举报

相关推荐

0 条评论