蓝桥杯 基础练习 2n皇后问题

问题描述

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

AC代码

#include <iostream>
#include <cmath>

using namespace std;
const int maxn = 10;
int n, d[maxn][maxn], ans = 0, p1[maxn], p2[maxn];
bool hash1[maxn] = {false}, hash2[maxn] = {false};

void dfs2(int index) { // 放白皇后
    if (index == n) {
        ans++;
        return;
    }
    for (int i = 0; i < n; i++) // 第i列
        if (!hash2[i] && d[index][i] == 1 && i != p1[index]) { // 判断该位置是否存在黑皇后
            bool flag = true;
            for (int j = 0; j < index; j++)
                if (abs(index - j) == abs(i - p2[j])) {
                    flag = false;
                    break;
                }
            if (flag) {
                p2[index] = i;
                hash2[i] = true;
                dfs2(index + 1); // 摆放下一行
                hash2[i] = false;
            }
        }
}

void dfs1(int index) { // 放黑皇后
    if (index == n) {
        dfs2(0);
        return;
    }
    for (int i = 0; i < n; i++) // 第i列
        if (!hash1[i] && d[index][i] == 1) { // 第i列还没有黑皇后并且可以摆放皇后
            bool flag = true;
            for (int j = 0; j < index; j++) // 遍历之前的皇后
                if (abs(index - j) == abs(i - p1[j])) {
                    flag = false;
                    break;
                }
            if (flag) {
                p1[index] = i;
                hash1[i] = true;
                dfs1(index + 1); // 摆放下一行
                hash1[i] = false;
            }
        }
}

int main() {
    cin >> n;
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            cin >> d[i][j];
    dfs1(0);
    cout << ans << endl;
    return 0;
}