题目1460:蓝桥杯基础练习VIP-2n皇后问题

#include<iostream>
#include<cmath>
using namespace std;
int vis[9][9];
int n,count=0;
bool check(int x,int y,int c){//c=2代表白皇后，c=3代表黑皇后
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(vis[i][j]==c){
if(i==x||j==y||abs(i-x)==abs(j-y)){
return false;
}
}
}
}
return vis[x][y]==1;
}
void dfs(int x,int c){
if(x==n){
if(c==3){
count++;
}else{
dfs(0,3);
return;
}
}
for(int i=0;i<n;i++){//在当前行，检查每一列的情况
if(check(x,i,c)){
int t=vis[x][i];
vis[x][i]=c;
dfs(x+1,c);
vis[x][i]=t;
}
}
}
int main(){
cin>>n;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
cin>>vis[i][j];
}
}
dfs(0,2);
cout<<count<<endl;
return 0;
}