问题描述
例题:n皇后
n皇后问题:输入整数n,要求n个国际象棋的皇后,摆在n*n的棋盘上,互相不能攻击(即n个皇后可以横着吃,斜着吃,竖着吃。),输出全部方案。
输入一个正整数N,则程序输出N皇后问题的全部摆法
输出结果里的每一行都代表一种摆法。行里的第i个数字如果是n,就代表第i行的皇后应该放在第n列。
皇后的行、列编号都是从1开始算。
样例输入
4
样例输出
2 4 1 3
3 1 4 2
算法思想
定义全局变量N,表示N个皇后要摆放,数组queenPos[100],表示最多是100个元素,第i行的皇后应该放在哪一列。行列号都是从0开始算的
定义一个函数,Nqueen(k)表示假定在第0行到k-1行都摆放好的情况下,我们接下来从k行开始,往后摆放皇后,直到把n个皇后全部放好。
我们在main里面从第0行开始摆放皇后,直到n个皇后都摆好
在Nqueen(k) 函数里,当这个函数被调用的时候,在0~k-1行皇后都已经摆好的情况下,摆第k行及其后的皇后。
我们首先判断k是否等于n,如果k等于n,表明n个皇后都摆好了,我们只用把摆好的皇后的位置输出即可。输出的结果里面从1开始算。
如果k不等于n,我们就逐个尝试第k个皇后的位置,第k个皇后有n个位置可以尝试,对每个可能的位置我们都要判断该位置是否可行,我们要判断第k个位置如果摆在列I,我们要考虑它跟前面摆好的从0到k-1个皇后的位置是否冲突。
即判断两个皇后位置是否会斜着或者相等。斜着的时候,行的差和列的差是相同的。注意,要取绝对值。如果冲突,就break,我们就不需要再看下一个皇后了。
当循环执行完了,我们就判断j是不是等于k,如果等于,就表明没有执行break,即不冲突,我们就可以把第k行的皇后摆放在位置i,然后我们就要继续到第k+1行开始摆放后续的皇后。
为什么nqueen(0)被调用的时候,最后就能输出所有解,因为对每一行的皇后,所有可能的位置我们都进行了尝试,与n重循环类似。
程序代码
#include<iostream>
#include<cmath>
using namespace std;
int N;
//用来存放放好的皇后位置,下标i表示第i行,其值表示列
int queenPos[100];
//nQueen(k)表示从0到k-1行的皇后都摆放好后,我们从第k行开始,找到后续皇后的摆放位置
void nQueen(int k){
int i,j;
//n个皇后都摆放好了,只需输出即可。
if(k==N){
for(int i=0;i<N;i++){
cout<<queenPos[i]+1<<" "; //输出结果下标从1开始
}
cout<<endl;
}else{
//逐个尝试第k个皇后的位置
for(i=0;i<N;i++){
//和已经摆好的k个皇后的位置进行比较,看是否冲突
for(j=0;j<k;j++){
//如果两列相等或者行和列的差值相等,则为冲突
if((i==queenPos[j])||(abs(i-queenPos[j])==abs(k-j))){
break;
}
}
//当前选的位置i不冲突
if(j==k){
//将第k个皇后摆放在位置i,并继续寻找后续阶段的皇后位置
queenPos[k]=i;
nQueen(k+1);
}
}
}
}
int main(){
cin>>N;
nQueen(0);
return 0;
}