2n皇后问题
#include<iostream>
#include<vector>
using namespace std;
int cnt = 0, n;
vector< vector<int> > vi(9, vector<int>(9,0) ); //储存棋盘
vector<int> row1(9, 0), row2(9, 0), md1(19, 0), md2(19, 0), vd1(19, 0), vd2(19, 0); // row1记录每行是否有放了黑棋, row2记录每行是否有放了白棋。 md1记录每条主对角是否有放了黑棋,md2记录每条主对角线是否有放了白棋。vd1记录每条副对角线是否有放了黑棋,vd2记录每条副对角线是否有放了白棋。
void DFS(int deep){ //回溯
if(deep == n+1){ //成功摆放完,cnt++
cnt++;
return ;
}
for(int i=1; i<=n; i++){
if(vi[deep][i] && !row1[i] && !md1[deep-i+n] && !vd1[deep+i]){ //判断此处是否可以摆放棋子,且所在行,对角线处是否有同颜色的棋子,列不用,因为每列就放该颜色一枚棋子
vi[deep][i]=0; row1[i]=1; md1[deep-i+n]=1; vd1[deep+i]=1; //修改数据
for(int j=1; j<=n; j++){ //处理另一种颜色的棋子
if(vi[deep][j] && !row2[j] && !md2[deep-j+n] && !vd2[deep+j]){
vi[deep][j]=0; row2[j]=1; md2[deep-j+n]=1; vd2[deep+j]=1;
DFS(deep+1);
vi[deep][j]=1; row2[j]=0; md2[deep-j+n]=0; vd2[deep+j]=0; //改回原来的数据,防止数据变脏
}
}
vi[deep][i]=1; row1[i]=0; md1[deep-i+n]=0; vd1[deep+i]=0;
}
}
}
int main(){
cin>>n;
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)
cin>>vi[i][j];
DFS(1);
cout<<cnt<<endl;
return 0;
}