2n皇后问题

2n皇后问题

---

每天一篇CSDN，快乐无极限！
如果有更好的解题方法，欢迎评论区讨论哦🙂

题目描述:
给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。

Input

输入的第一行为一个整数n，表示棋盘的大小。
接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。

Output

输出一个整数，表示总共有多少种放法。

Sample Input

4
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);//输出
}

完啦