基础训练 2n皇后问题

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;
}