（蓝桥杯基础训练题目）2n皇后（回溯法）

如下：

#include <iostream>
#include<stdlib.h>
#include<vector>
#include<math.h>
using namespace std;
int warray[9] = { 0 }; //用于存放白皇后位置 warray[1] = 2,表示1行2列有白皇后
int barray[9] = { 0 }; //用于存放黑皇后位置
int vis[9][9] = { 0 }; //用于存放输入矩阵
int n;
int cnt = 0; //记录摆放方法总数
int whiteQuene(int x) { //摆放白皇后
	if (x > n) {
		cnt++;
		return 0;
	}

	for (int tj = 1; tj <= n; tj++) {
		if (!vis[x][tj] || tj == barray[x]) {//此处若放有黑皇后，那么不能在这放置
			continue;
		}
		int j = 1;
		while (j < x) {  //检测是否相同列或者对角线已存在相同颜色皇后
			if (warray[j] == tj || fabs(j - x) == fabs(warray[j] - tj)) {//这里取绝对值直接将对半分的两条对角线全部排除
				break;
			}
			j++;//保存此时的j值，如果检测到不是本行放置的皇后，那么不再进入递归
		}
		if (j == x) {
			warray[x] = tj;
			whiteQuene(x + 1);
		}
	}
	return 0;

}
int blackQuene(int x) { // 摆放黑皇后
	if (x > n) {
		whiteQuene(1); //每当黑皇后摆放完一种，就开始摆放白皇后
		return 0;
	}
	for (int tj = 1; tj <= n; tj++) {
		if (!vis[x][tj]) {
			continue;
		}


		int j = 1;
		while (j < x) { //检测是否相同列或者对角线已存在相同颜色皇后
			if (barray[j] == tj || fabs(x - j) == fabs(tj - barray[j])) {
				break;
			}
			j++;
		}
		if (j == x) {
			barray[x] = tj;
			blackQuene(x + 1);
		}
	}
}
int main()
{
	cin >> n;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++) {
			int t;
			cin >> t;
			vis[i][j] = t;
		}
	}
	blackQuene(1);
	cout << cnt;
	return 0;
}