蓝桥 - 2n皇后问题 搜索

思路：按n皇后思路 两次搜索即可 注意黑白皇后也不能重合

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#define ll long long
#define inf 0x3f3f3f3f
#define sd(a) scanf("%d",&a)
#define sdd(a,b) scanf("%d%d",&a,&b)
#define cl(a,b) memset(a,b,sizeof(a))
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define sddd(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define dbg() printf("aaa\n")
using namespace std;
int n;
int a[10],b[10];
int mp[10][10];
int num;
int num2;
void dfs2(int i){
    if(i==n+1){
        num2++;
    }
    rep(j,1,n){//对于每个位置
        if(!mp[i][j]) continue;
        if(a[i]==j) continue;
        bool flag=true;
        rep(k,1,i-1){
            if(b[k]==j||abs(b[k]-j)==abs(k-i)){
                flag=false;
                break;
            }
        }
        if(flag) {
            b[i]=j;
            dfs2(i+1);
        }
    }
}
void dfs1(int i){//对于当前这一行
    if(i==n+1){
        num2=0;
        dfs2(1);
        num+=num2;
        return;
    }
    rep(j,1,n){//对于每个位置
        if(!mp[i][j]) continue;
        bool flag=true;
        rep(k,1,i-1){
            if(a[k]==j||abs(a[k]-j)==abs(k-i)){
                flag=false;
                break;
            }
        }
        if(flag) {
            a[i]=j;
            dfs1(i+1);
        }
    }
}
int main() {
    num=0;
	sd(n);
    rep(i,1,n){
        rep(j,1,n){
            sd(mp[i][j]);
        }
    }
    dfs1(1);
    printf("%d\n",num);

	return 0;
}
/*
*/

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

接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。
输出
输出一个整数，表示总共有多少种放法。
样例输入

4

1 1 1 1

1 1 1 1

1 1 1 1

1 1 1 1

样例输出

2