题干:
在 n*n 格(n<=8)的国际象棋棋盘上摆放 n 个皇后,使其不能互相攻击,即任意两个皇后都不能在同一行、同一列或同一斜线上,问有多少种摆法。
分析:
使用回溯法,从第一行开始一行一行从左往右放皇后,若冲突则往右判断,若当前行能放则进入下一行,不能放则回退到上一行,采用递归方式,递归函数参数传递行号,成功出口是行数大于n。数据存放用一维数组即可,数组下标表示行号,对应位置的值表示列号。冲突判断也用一维数组,设行号为i,列号为j,则列判断下标为j,右上到左下判断下标为i+j,左上到右下判断下标为i+8-j。
题解:
#include"bits/stdc++.h"
using namespace std;
int ruToLd[2*8+1],luToRd[2*8+1],col[8+1],queen[8+1],num=0;
void printAns(int n){
for(int y=1;y<=n;++y){
for(int x=1;x<=n;++x){
if(queen[y]==x) cout<<"Q ";
else cout<<"# ";
}
cout<<endl;
}
cout<<endl;
}
void backtrack(int i,int n){
if(i>n){
num++;
printAns(n);
}
else{
for(int j=1;j<=n;++j){
if((col[j]==0)&&(ruToLd[i+j]==0)&&(luToRd[i+8-j]==0)){
queen[i]=j;
col[j]=1,ruToLd[i+j]=1,luToRd[i+8-j]=1;
backtrack(i+1,n);
col[j]=0,ruToLd[i+j]=0,luToRd[i+8-j]=0;
}
}
}
}
int main(){
int n;
cin>>n;
backtrack(1,n);
cout<<num;
return 0;
}