题目
在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
你的任务是,对于给定的N,求出有多少种合法的放置方法。
Input
共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。
Output
共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
Sample Input
1
8
5
0
Sample Output
1
92
10
思路
经典的回溯思想的练习,本题即是是非学过回溯算法的分界线
使用递归实现dfs(回溯)
使用数组a的下标表示棋盘的行,然后数组值代表棋子在第几行第几列
写一个检查的函数,因为行是遍历的所以棋子比定不在同一行,只需判断棋子是不是在同一列,是不是在同一斜线上即可
(abs(a[i]-c))==abs(i-r)代表本次放置的棋子和上一个放置的棋子在同一直线上,因为同一斜线上有四种可能性,简单模拟归纳一下即可发现用绝对值就可以代表是否在同一斜线上。
代码
#include"bits/stdc++.h"
using namespace std;
int tot=0,y;
int a[12]={0};
bool check (int c,int r){
for(int i=0;i<r;i++){
if(a[i]==c||(abs(a[i]-c))==abs(i-r))
return false;
}
return true;
}
void dfs(int now){
if(now==y){
tot++;
return;
}
for(int i=0;i<y;i++){
if(check(i,now)){
a[now]=i;
dfs(now+1);
}
}
}
int main(){
int ans[12]={0};
for(y=0;y<=10;y++){
memset(a,0,sizeof(a));
tot=0;
dfs(0);
ans[y]=tot;
}
while(cin>>y){
if(y==0)return 0;
cout<<ans[y]<<endl;
}
return 0;
}