蓝桥杯试题 基础练习 2n皇后问题（n皇后问题/八皇后问题）【dfs+回溯】

试题 基础练习 2n皇后问题

题目链接

资源限制

限制：1.0s 内存限制：512.0MB

问题描述

给定一个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

样例输入

4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1

样例输出

0

思路：

经典的dfs问题，八皇后问题的拓展
设两个数组分别记录某一行黑皇后和白皇后的列位置
从第1行到第n行递归dfs
每行每列都要判断当前位置是否能放

完整代码：

using namespace std;

int maze[10][10];
int n;
int sum=0;//总数
int ind_b[10];//记录黑在第x行的y
int ind_w[10];//记录白在第x行的y

bool can_b(int x,int y){
    if(maze[x][y]==0)
        return false;
    for(int i=1;i<x;i++){//判断前x行就可以了
        if(ind_b[i]==y||i-x==ind_b[i]-y||x-i==ind_b[i]-y)//是否同一列，左右对角线
            return false;
    }
    return true;
}

bool can_w(int x,int y){
    if(maze[x][y]==0)
        return false;
    for(int i=1;i<x;i++){
        if(ind_w[i]==y||i-x==ind_w[i]-y||x-i==ind_w[i]-y)
            return false;
    }
    return true;
}

void dfs_w(int x){
    if(x==n+1){
        sum++;
        return;
    }
    for(int y=1;y<=n;y++){
        if(can_w(x,y)){
            ind_w[x]=y;
            maze[x][y]=0;
            dfs_w(x+1);
            ind_w[x]=0;
            maze[x][y]=1;
        }
    }
}
void dfs_b(int x){//先放黑皇后
    if(x==n+1){
        dfs_w(1);//放完后再放白皇后
        return;//退出
    }
    for(int y=1;y<=n;y++){
        if(can_b(x,y)){
            ind_b[x]=y;//记录
            maze[x][y]=0;//标记已有棋子
            dfs_b(x+1);
            //回溯
            ind_b[x]=0;
            maze[x][y]=1;
        }
    }
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            cin>>maze[i][j];
    dfs_b(1);
    cout<<sum;
    return 0;
}