题目概述: 在N×N格的国际象棋上摆放N个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法。
思路:对于此问题,可以将每列皇后的行号列举出来,则问题转化为1~n的一个全排列。
代码一(暴力法):
#include<stdio.h>
#include<stdbool.h>
#include<math.h>
int n,p[10],hashtable[10]={false};
int count=0;
int generatep(int index){
if(index==n+1) { //表示已经排列完毕,接下来判断是否在同一条对角线上就可以
bool flag=true;
for(int i=1;i<=n;i++){ //暴力枚举 ,遍历任意两个皇后
for(int j=i+1;j<=n;j++){
if(abs(i-j)==abs(p[i]-p[j])){ //同一对角线上 :行差等于列差
flag=false;
}
}
}
if(flag) count++;
return count;
}
for(int x=1;x<=n;x++){ //枚举1-n,试图将x填入p[index]
if(hashtable[x]==false){ //如果x不在p[1]-p[index]中
p[index]=x; //将x填入到p[index]位置
hashtable[x]=true; //进行标记
generatep(index+1); //处理后续的index+1
hashtable[x]=false; //处理完 p[index]为x的子问题后 ,对hashtable复原
}
}
}
int main()
{
printf("Input the Size of the Chess:");
scanf("%d",&n); //问题规模
generatep(1);
printf("Answer Number:%d\n",count);
return 0;
}
代码二(回朔法):
#include<stdio.h>
#include<stdbool.h>
#include<math.h>
int n,p[10],hashtable[10]= {false};
int count=0;
int generatep(int index)
{
if(index==n+1)//递归边界,生成一个合法方案
{
count++; //能到达这一步的都是合法的方案
return count;
}
for(int x=1; x<=n; x++)//第x行
{
if(hashtable[x]==false) //第x行还没有皇后
{
bool flag=true;//flag为true表示当前皇后和之前的皇后不会冲突
for(int pre=1; pre<index; pre++) //遍历之前的皇后,第index列皇后行号为x,第pre列皇后的行号为p[pre]
{
if(abs(index-pre)==abs(x-p[pre])) //处于同一对角线
{
flag=false;
break;
}
}
if(flag==true)//如果可以放置到同一对角线
{
p[index]=x; //将x填入到p[index]位置
hashtable[x]=true; //进行标记
generatep(index+1); //处理后续的index+1
hashtable[x]=false; //处理完 p[index]为x的子问题后 ,对hashtable复原
}
}
}
}
int main()
{
printf("Input the Size of the Chess:");
scanf("%d",&n);
generatep(1);
printf("Answer Number:%d\n",count);
return 0;
}
ps:回朔法可以避免很多不必要的计算。比如在放置完一部分皇后以后,可能剩余的皇后无论怎么放置都不合法,则没必要继续递归了,直接返回上一层即可。