在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
你的任务是,对于给定的N,求出有多少种合法的放置方法。
Input
共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。
Output
共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
Sample Input
1
8
5
0
Sample Output
1
92
10
关键:
- 如何递归
- 如何剪枝和回溯
细节:
- 打表。在 main() 中提前算出从 1 到 10 的所有 N 皇后 问题的答案,并储存在数组中,等读取输入后立即输出。如果不打表,而是等输入 N 后再单独计算输出,会超时。
- 递归搜索 DFS() 。递归程序十分简洁,在一行中寻找合适位置放置皇后,然后递归放置其他行的皇后,直到放完。
- 回溯判断。判断新放置的皇后和已放置的皇后在横向、纵向、斜对角方向是否冲突。其中,横向不需要判断,因为在递归的过程中已经是按不同行放置。
- 复杂度。在上述程序中, DFS() 一行行放置皇后,复杂度为 O(N!) ;检查冲突,复杂度为 O(N) ;总复杂度为 O(N * N!) 。因此,对于 N > 11 的 N 皇后问题,需要用新的方法 。
代码:
#include<iostream>
#include<cmath>
using namespace std;
int n, num, visit[11];
void dfs(int x) {
if (x == n) {//放置了 n 行皇后
num++;
return;
}
for (int i = 0; i < n; i++) {
int j = x;
for(j=0;j<x&&visit[j]!=i&&abs(x-j)!=abs(i-visit[j]);j++){}//检测函数
if (j == x) {
visit[x] = i;//在第 x 行,i 列放置皇后
dfs(x + 1);//继续放下一行皇后
}
//若该点不能放置,不作为,继续尝试该行下一点
visit[x] = NULL;//记得删除放置标记
}
}
int main() {
int ans[12] = { 0 };
for (n = 1; n <= 10; n++) {//打表
num = 0;
dfs(0);
ans[n] = num;
}
while (cin >> n&&n){
cout << ans[n] << endl;
}
}