843. n-皇后问题
题目
n-皇后问题是指将 n 个皇后放在 n∗n 的国际象棋棋盘上,使得皇后不能相互攻击到,即任意两个皇后都不能处于同一行、同一列或同一斜线上。
现在给定整数n,请你输出所有的满足条件的棋子摆法。
输入格式
共一行,包含整数n。
输出格式
每个解决方案占n行,每行输出一个长度为n的字符串,用来表示完整的棋盘状态。
其中”.”表示某一个位置的方格状态为空,”Q”表示某一个位置的方格上摆着皇后。
每个方案输出完成后,输出一个空行。
输出方案的顺序任意,只要不重复且没有遗漏即可。
数据范围
1 ≤ n ≤ 9 1≤n≤9 1≤n≤9
输入样例:
4
输出样例:
.Q..
...Q
Q...
..Q.
..Q.
Q...
...Q
.Q..
本题就是经典的n皇后问题,在n*n的棋盘上放置n个皇后,要求是任意两个皇后不能在同一行、同一列或同一对角线上,输出所有满足条件的摆法。方法是用dfs,可以通过两种思路来考虑。
思路一(按行枚举)
搜索全排列。首先,生成1个全排列后判断有没有两个皇后在同一行、同一列、同一对角线上。当把n个全排列生成出来后再判断这个摆放方案是否是合法的。因为在棋盘上每一行都放一个且只能放一个皇后,可以按顺序一行行放皇后,枚举每一行皇后放在哪一位置(列)。
列出全排列的n个空位,第 i 空位上的数为 j 表示皇后在第 i 行的第 j 列。如果第1行把皇后放在第1列,那么,第1个空位上就是1;第2行把皇后放在第3列,那么第2个空位上就是3。假设当前位置枚举某个数时,第3个空位枚举了一个4,就直接判断在第3行第4列放一个皇后有没有冲突,如果有冲突就不向下走了,就是剪枝。就是提前判断当前方案一定是不合法的,没必要再往下搜了,直接剪枝、回溯。
注意剪枝。
代码
#include <iostream>
using namespace std;
const int N = 10;
int n;
char g[N][N];
bool col[N], dg[N], udg[N];
//三个布尔数组,分别来判断列上、对角线上、反对角线上是否冲突
void dfs(int u) {
if(u == n) {
//u == n表示已经搜了n行,所以输出这个方案