2n皇后问题
每天一篇CSDN,快乐无极限!
如果有更好的解题方法,欢迎评论区讨论哦🙂
题目描述:
给定一个n*n的棋盘,棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后,使任意的两个黑皇后都不在同一行、同一列或同一条对角线上,任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法?n小于等于8。
输入的第一行为一个整数n,表示棋盘的大小。
接下来n行,每行n个0或1的整数,如果一个整数为1,表示对应的位置可以放皇后,如果一个整数为0,表示对应的位置不可以放皇后。
输出一个整数,表示总共有多少种放法。
Sample Input4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1
Sample Output
2
这道题可以参考我的另一篇博客,八皇后问题。因为要放置黑白两种皇后,所以我们可以写两次dfs,dfs1放黑,dfs2放白,dfs2是套在dfs1里的。先进行dfs1,如果第dfs1可以找到黑皇后的一种摆法,则在当前地图接着dfs2,如果dfs2也找到了一种,那么总的total加一。这样就把所有的情况给搜了出来。可以自己看一下下面的代码。
废话不多说,上代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;
int a[100],b[100]={0},c[100]={0},d[100]={0};
//数组a记录结果的列号,数组b记录列占领,
//数组c记录左下到右上到对角线占领,数组d记录右下到左上的对角线占领。
int total=0,n;
void printf(){//输出函数
if(total<=2)
{
for(int k=1;k<=n;k++)
printf("%d ",a[k]);
printf("\n");
}
total++;
}
void dfs(int i){
if(i>n){
printf();
return;
}
else{
for(int j=1;j<=n;j++){
if((!b[j])&&(!c[i+j])&&(!d[i-j+n]))
{
a[i]=j;//记录当前行所对应的列
b[j]=1;//宣布占领此列
c[i+j]=1;//宣布占领两条对角线
d[i-j+n]=1;
dfs(i+1);//进行下一轮深搜,放置下一个皇后
b[j]=0;
c[i+j]=0;
d[i-j+n]=0;
//回溯
}
}
}
}
int main(){
scanf("%d",&n);
dfs(1);
printf("%d",total);
}
#include<cstdio>
#include<iostream>
using namespace std;
int map[9][9];//地图
int n;
int b1[100]={0},c1[100]={0},d1[100]={0};
int b2[100]={0},c2[100]={0},d2[100]={0};
//b1,c1,d1记录黑皇后,b2,c2,d2记录白皇后
//数组b记录列占领,
//数组c记录左下到右上到对角线占领,数组d记录右下到左上的对角线占领。
int total=0;//初始化total
void dfs2(int k){//dfs2找白皇后
if(k>n)
{
total++;//如果白皇后也找到,则total++;
return;
}
else{
for(int j=1;j<=n;j++){
if((!b2[j])&&(!c2[k+j])&&(!d2[k-j+n])&&map[k][j]==1){
b2[j]=1;//宣布占领
c2[k+j]=1;
d2[k-j+n]=1;
map[k][j]=0;//标记已经有棋子
dfs2(k+1);//找下一行
b2[j]=0;
c2[k+j]=0;
d2[k-j+n]=0;
map[k][j]=1;
//回溯
}
}
}
}
void dfs1(int i){//先用dfs1找黑皇后
if(i>n)
{
dfs2(1);//如果黑皇后深搜找到一种摆法,则dfs2继续找白皇后
}
else{
for(int j=1;j<=n;j++){
if((!b1[j])&&(!c1[i+j])&&(!d1[i-j+n])&&map[i][j]==1){
b1[j]=1;//宣布占领
c1[i+j]=1;
d1[i-j+n]=1;
map[i][j]=0;//标记有棋子
dfs1(i+1);//寻找下一行
b1[j]=0;
c1[i+j]=0;
d1[i-j+n]=0;
map[i][j]=1;
//回溯
}
}
}
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%d",&map[i][j]);
dfs1(1);//先进行dfs1找黑皇后
printf("%d",total);//输出
}
完啦