将n个queen放在一个n*n的棋盘上,要求同一行,同一列,两斜线不能放。
这就是个简单的回溯问题。
但是如何判断能不能放,这是个难点。
存疑:
当时想着开个二维数组来判断这一个坐标可不可以放。
但是这里有个问题,两个queen(1和2)的斜线可能有焦点,那我们回溯的时候会把这个坐标重置(从queen 2回溯回去)。但是这样的话queen 1的斜线没有了作用。
2020.9.12
二维数组可以放。
num[i][j]=1 说明放了
num[i][j]=0,没放皇后,且可以放
num[i][j]<0; 这个位置不可以放皇后
先把放置皇后的行,列,斜线出全部num- -,变为负数
然后遍历下一个可以放得位置,进行搜索。
最终ans++,且把之前的全部num++即可
也可以开四个一维数组。
b[i]=j,表示棋盘的第i行第j列上有queen
c[j]=1,表示第j列上没有queen
d[k]=1表示第k条下行(向右下方向)对角线上无queen。k=i+j.就是直线啊
u[k]=1表表弟k条(左下)无queen,k=i-j
try(s){//s 行数
j=0;//列数
q=0;//为0,放置未成功
while(!q&&j<n){
j++;
if(Safe(s,j)){
record(s,j);
if(s==n){//
q=1;
output();
}else try(s+1);
if(!q) move-off(s,j);//这一个位子不能放queen,
//why??因为在这一个位置递归下去,都不能再找到满足答案的解。
}
}
}
void record(s,j){
b[s]=j;
c[j]=0;
d[s+j-1]=0;
u[s-j+n-1]=0;
}
void move-off(s,j){
b[s]=0;
c[j]=1;
d[s+j-1]=1;
u[s-j+n-1]=1;
}
boolean safe(s,j){//行列
if(c[j]==1&&d[s+j-1]==1&&u[s-j+n-1]==1) return true;
return false;
}
queen是从第0行开始放的且一行只能有一个queen,所以第i个queen也就是第i行。
用一个数组flag来表示。flag[i]=j
表明第i行的queen放在了第j列。
这样,考虑斜线的时候可以用两点直线的斜率(Y1-Y2/X1-X2)即可
#include<iostream>
#include<set>
#include<vector>
using namespace std;
int ans=0;
int flag[100];//flag[i]=j, 第i个皇后的列数j
bool place(int col,int row){
for(int i=0;i<row;i++){ //找出已经放置的queen的行,判断他们与将要放的queen是否冲突
if(flag[i]==col||abs(i-row)==abs(col-flag[i])) return false;
}
return true;
}
void dfs(int n,int row){
if(row==n){
ans++;
return;
}
for(int i=0;i<n;i++){//列数
if(place(i,row)){//判断第row行,第i列可不可以放queen
flag[row]=i;
dfs(n,row+1);
//flag[row]=0;
//发现去掉这里也没事,原因:我们先判断再放,所以回溯也没事。
}
}
}
int main(void){
int n=4;
dfs(n,0);
cout<<ans;
}