原题:传送门
题意: 给n x n的棋盘,放n个皇后,要求任何两个皇后不能在同一行或同一列或同一对角线上,按字典序输出所有方案
思路: 用ans数组存储第i行皇后所在的列,这样就不存在两个皇后在同一行的情况了;用vis[i]记录第i列是否存在皇后,这样就不存在两个皇后在同一列的情况了;再写个check(int m)检查当前摆法的对角线的合法性,dfs搜索就完事了
友情提示: dfs()返回后,一定要把被置为true的vis[i]还原为false
Code:
#include<iostream>
using namespace std;
int n;
int ans[17];//第i行存储皇后所在位置ans[i]
bool vis[17];//vis[i]表示数字i是否被使用过,也就是第i列是否被摆放了皇后
bool is_print = false;
bool check (int m) {
for (int i=m+1;i<=n;i++)//只判断前m个位置,故把m+1到n得数值清零
ans[i] = 0;
for (int i=1;i<=m;i++) {
for (int j=1;j<=ans[i] - 1;j++) {
if (ans[i] - j == ans[i - j] || ans[i] - j == ans[i + j]) {
return false;
}
}
for (int j=1;j+ans[i]<=m;j++) {
if (ans[i] + j == ans[i - j] || ans[i] + j == ans[i + j]) {
return false;
}
}
}
return true;
}
void dfs (int m) {
if (m-1 == n) {//因为从1开始的dfs,所以取n个值时,m已经等于n+1了
is_print = true;
for (int i=1;i<=n;i++) {
cout<<ans[i];
if (i != n) {
cout<<" ";
} else {
cout<<endl;
}
}
}
for (int i=1;i<=n;i++) {
if (!vis[i]) {
ans[m] = i;
if (!check(m)) {//当前m个都不满足时,就没必要继续了,实现了所谓的剪枝
ans[m] = 0;
continue;
}
vis[i] = true;
dfs(m+1);
vis[i] = false;
}
}
}
int main()
{
cin>>n;
fill(vis, vis + 17, false);
dfs(1);
if (!is_print) {
cout<<"no solute!"<<endl;
}
return 0;
}