[蓝桥杯] BASIC-27 基础训练 【2n皇后问题】 Apare_xzc 100
题面：
分析：
同八皇后问题
我的AC代码
#include <bits/stdc++.h>
#define BLACK 2
#define WHITE 3
using namespace std;
int a[10][10],n,ans;
int vb[10],vw[10]; //记录每一行黑白的列数
bool visBlack[10],visWhite[10]; //记录第i行的皇后所在的列数
void dfs(int);
int main()
{
cin>>n;
for(int i=0;i<n;++i)
for(int j=0;j<n;++j)
scanf("%d",&a[i][j]);
ans = 0;
dfs(0);
cout<<ans<<endl;
return 0;
}
void dfs(int x)
{
if(x==n) //搜到结果了，答案加一
{
ans++;return;
}
for(int black=0;black<n;black++) //枚举第x行黑皇后的列位置
{
for(int white=0;white<n;++white) //枚举第x行白皇后的列位置
{
if(black==white||a[x][black]==0||a[x][white]==0) continue;
//标0的地方不能放，黑白皇后不能放一起
if(visBlack[black]||visWhite[white]) continue;
//不能和之前放的同色皇后在同一列
bool ok = true;
for(int i=0;i<x&&ok;++i) //每一行的黑白
{ //判断对角线
if(abs(vb[i]-black)==abs(i-x)||abs(vw[i]-white)==abs(i-x))
ok = false;
}
if(!ok) continue;
visBlack[black] = true, visWhite[white] = true; //标记
a[x][white] = WHITE, a[x][black] = BLACK; //放置
vb[x] = black, vw[x] = white; //记录位置
dfs(x+1); //搜索
visBlack[black] = false, visWhite[white] = false; //清除标记
a[x][white] = 1, a[x][black] = 1; //还原棋盘
}
}
}
