洛谷P1562 还是N皇后
原文链接
(https://www.luogu.com.cn/problem/P1562)
题目描述
正如题目所说,这题是著名的N皇后问题。
输入格式
第一行有一个N。接下来有N行N列描述一个棋盘,“*”表示可放“.”表示不可放。
输出格式
输出方案总数
样例 #1
样例输入 #1
4
**.*
****
****
****
样例输出 #1
1
提示
0<n<=14;
题解
思路
- 先对输入数据的每一行进行状态压缩,我们用数组 row[x] 表示 第x行的每一列是否可以放置,如果此位置可放用0表示(输入中“*”的位置),不可放则用1表示(输入中“.”的位置)。
- dfs(d ,col ,lv ,rv) 中d表示当前行,col表示当前行状态,lv表示左对角线状态,rv表示右对角线状态。
- 为获取当前行哪个位置可以放皇后,先对上一步的状态进行取反;再将四者进行或运算,如此:all&(~(row[x] | col | lv | rv))
- 再使用 lowbit() 获取本行状态的二进制中的最后一个1,并删除它。
- 当对下一行状态进行处理时,由于皇后还会影响到其对角线上的位置,所以到右下对角线是需要 左移一位 ,到左下时需要 右移一位 。
- 当每一行都有皇后时,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;
}