回溯剪枝：n皇后问题

描述

n皇后问题：一个n×n的棋盘，在棋盘上摆n个皇后，满足任意两个皇后不能在同一行、同一列或同一斜线上的方案有多少种？

输入

第一行包含一个整数n。

输出

输出一个整数，表示方案数。

样例1输入

4

样例1输出

2

代码实现

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

// ================= 代码实现开始 =================

/* 请在这里定义你需要的全局变量 */

// 一个n×n的棋盘，在棋盘上摆n个皇后，求满足任意两个皇后不能在同一行、同一列或同一斜线上的方案数
// n：上述n
// 返回值：方案数

int allOne, ans;
// pos: 列信息，第几位bit位为1就表示该列已有皇后
// left: 左对角线信息，第几位bit位为1就在上一行该对角线已有皇后
// right: 右对角线信息，第几位bit位为1就在上一行该对角线已有皇后
void dfs(int pos, int left, int right){
	if(pos == allOne){ //当所有列都为1时，表示已被填满
		++ans;
		return;
	}
	//t代表可以放的集合对应的二进制数
	for(int t=allOne&(~(pos | left | right)); 0<t; ){
		int p = t & (-t);//low bit：二进制中最右边的1
		dfs( pos|p, (left|p)<<1, (right|p)>>1);//左对角线每次左移，右对角线每次右移
		t ^= p;//对刚刚置1的位进行剪枝
	}	
}

int getAnswer(int n) {
    /* 请在这里设计你的算法 */
	ans = 0;
	allOne = (1<<n)-1;
	dfs( 0, 0, 0);
	return ans;
}

// ================= 代码实现结束 =================

int main() {
    int n;
	cin >> n;
	cout << getAnswer(n) << endl;
	return 0;
}

来源

邓俊辉老师的《算法训练营》第一期。