蓝桥杯——2n皇后问题

题目来源：蓝桥杯基础练习

知识点：回溯，递归

问题描述
　　给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。
输入格式
　　输入的第一行为一个整数n，表示棋盘的大小。
　　接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。
输出格式
　　输出一个整数，表示总共有多少种放法。
样例输入
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
2
样例输入
4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
0

题目分析

这是 n皇后问题的变体。在我看来，本题有两种解题思路：

1. 逐行放置，每次同时放置黑、白皇后。
2. 先放置黑皇后，完成后再放置白皇后。

第一种解法在分析时相对复杂一点（可以尝试一下），这里我使用第二种解法。

首先要弄明白皇后放置的规则：一个皇后一旦确定位置，与它同一行、同一列，以及两条对角线上所有的位置都不能再放皇后了。下图简单示意了一下。

接着开始放置皇后。我们可以先把问题简化，思考如何放置一种皇后。这里以黑皇后为例。这样就回到了 n皇后的问题。我们从第一行开始放置，一行一行地寻找合法的位置。这里可以看出，每次放置时的动作都是相似的：定位行，找合法列，放置。因此，可以使用递归来编写代码。

行数可以作为递归的参数传入。这里的问题在于如何判断一个位置是否合法。我们可以总结出一个位置有如下几种状态：

• 状态1：空但不能放置皇后。
• 状态2：空且可以放置任意颜色的皇后。
• 状态3：已经放置了皇后，格子被占。
• 状态4：不能放置白皇后。
• 状态5：不能放置黑皇后。

那么就可以定义一些数组，用来标记一个位置的不同状态。这里我规定先黑后白的顺序，放置黑皇后时一个位置合法的条件是：!(状态1) && !(状态5)。放置白皇后时一个位置合法的条件是：!(状态1) && !(状态3) && !(状态4)。

别忘了递归的结束条件！当放置黑皇后填完最后一行时，就从第一行开始放置白皇后。当白皇后放置完最后一行时，就求出了一种解法，可以回溯到上一步了。

代码

#include <bits/stdc++.h>
using namespace std;

const int M = 10;
int board[M][M];
int flagb[M][M];
int flagw[M][M];
int pos[M][M];
int n;
int ans;

void backtrack2(int i);

//放黑皇后 
void backtrack1(int i) {
	if(i == n) {
		backtrack2(0);
		return;
	} 
	
	for(int j=0; j<n; j++) {
		if(!board[i][j] || flagb[i][j]) continue;
		pos[i][j] = 1; // 放置标记
		for(int k=0; k<n; k++) {
			flagb[k][j] += 1;
			if(k-i+j < n) flagb[k][k-i+j] += 1;
			if(j-k+i >= 0) flagb[k][j-k+i] += 1;
		} 
		backtrack1(i + 1); // 回溯 
		pos[i][j] = 0;
		for(int k=0; k<n; k++) {
			flagb[k][j] -= 1;
			if(k-i+j < n) flagb[k][k-i+j] -= 1;
			if(j-k+i >= 0) flagb[k][j-k+i] -= 1;
		} 
	}
}
void backtrack2(int i) {
	if(i == n) {
		ans++;
		return;
	} 

	for(int j=0; j<n; j++) {
		if(!board[i][j] || pos[i][j] || flagw[i][j]) continue;
		pos[i][j] = 1; // 放置标记
		for(int k=0; k<n; k++) {
			flagw[k][j] += 1;
			if(k-i+j < n) flagw[k][k-i+j] += 1;
			if(j-k+i >= 0) flagw[k][j-k+i] += 1;
		} 
		backtrack2(i + 1); // 回溯
		pos[i][j] = 0;
		for(int k=0; k<n; k++) {
			flagw[k][j] -= 1;
			if(k-i+j < n) flagw[k][k-i+j] -= 1;
			if(j-k+i >= 0) flagw[k][j-k+i] -= 1;
		} 
	}
}

int main() {
	cin >> n; 
	
	for(int i=0; i<n; i++) {
		for(int j=0; j<n; j++) {
			cin >> board[i][j];
			flagb[i][j] = 0;
			flagw[i][j] = 0;
			pos[i][j] = 0;
		}
	}
	
	ans = 0;
	backtrack1(0);
	
	cout << ans << endl;
	
	return 0;
}