参考自:这位博主~
n皇后问题:输入整数n, 要求n个国际象棋的皇后,摆在n*n的棋盘上,互相不能攻击(即不能在同一行,同一列,同一斜线),输出全部方案。
输入一个正整数N,则程序输出N皇后
问题的全部摆法。
输出结果里的每一行都代表一种摆法。行里的第i个数字如果是n,就代表第i行的皇后应该放在第n列。
皇后的行、列编号都是从1开始算。
样例输入:
4
样例输出:
2 4 1 3
3 1 4 2
解题思路:
用一个数组C[]存放可能的结果数,每次判断当前列能不能放在C[cur]中(如果当前的列不和之前的列重合,不在同一斜线上,就可以放~),最后一直放到第n列说明满足题意,种类数加一。
#include<iostream>
#include<cstdio>
using namespace std;
int n,cnt=0;
int C[10010];
void dfs(int cur){
if(cur==n){
cnt++;
return;
}
for(int i=0;i<n;i++){
int flag=0;
C[cur]=i;
for(int j=0;j<cur;j++){
if(C[j]==C[cur]||(C[j]-j==C[cur]-cur)||(C[j]+j==C[cur]+cur)){//[2,2]或[2,3]或[2,1]
flag=1;
break;
}
}
if(!flag){
dfs(cur+1);
}
}
}
int main(int argc, char** argv) {
scanf("%d",&n);
dfs(0);
printf("%d\n",cnt);
return 0;
}