洛谷P1562 还是N皇后（位运算优化）————搜索蒟蒻

洛谷P1562 还是N皇后

原文链接

(https://www.luogu.com.cn/problem/P1562)

题目描述

正如题目所说，这题是著名的N皇后问题。

输入格式

第一行有一个N。接下来有N行N列描述一个棋盘，“*”表示可放“.”表示不可放。

输出格式

输出方案总数

样例 #1

样例输入 #1

4
**.*
****
****
****

样例输出 #1

1

提示

0<n<=14;

题解

思路

1. 先对输入数据的每一行进行状态压缩，我们用数组 row[x] 表示 第x行的每一列是否可以放置，如果此位置可放用0表示（输入中“*”的位置），不可放则用1表示（输入中“.”的位置）。
2. dfs(d ,col ,lv ,rv) 中d表示当前行，col表示当前行状态，lv表示左对角线状态，rv表示右对角线状态。
3. 为获取当前行哪个位置可以放皇后，先对上一步的状态进行取反；再将四者进行或运算，如此：all&(~(row[x] | col | lv | rv))
4. 再使用 lowbit() 获取本行状态的二进制中的最后一个1，并删除它。
5. 当对下一行状态进行处理时，由于皇后还会影响到其对角线上的位置，所以到右下对角线是需要 左移一位 ，到左下时需要 右移一位 。
6. 当每一行都有皇后时，ans++。

代码

#include <bits/stdc++.h>
using namespace std;
int n, ans, all;
int row[20];
int lowbit(int x){//寻找二进制中的最后一个1
	return x & (-x);
}
void dfs(int d, int col, int lv, int rv){
	if(d == n){//当每一行都有皇后时
		ans++;
		return ;
	}
	int vis = all & (~(row[d + 1] | col | lv | rv));//或运算+取反
	for(int i = vis; i; i -= lowbit(i)){//删除最后一个1
		int x = lowbit(i);
		dfs(d + 1, col + x, (lv + x)>>1, (rv + x)<<1);//进行下一行操作
	}
}
int main(){
	char c;
	cin >> n;
	all = (1<<n) - 1;
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= n; j++){
			cin >> c;
			if(c == '.'){//输入数据状态压缩
				row[i] |= (1<<(n - j));
			}
		}
	}
	dfs(0, 0, 0, 0);
	cout << ans;
	return 0;
}