解题思路:
蓝桥杯不错的一道暴力题,直接两个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;
}