2n皇后问题

 

上代码：

 

 

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;

int num,cot = 0;
int black_Queen[8];
int white_Queen[8];
int vis[8][8];
bool Check_Black(int n)
{
    for(int i = 0;i < n;++i)
    {
        if(black_Queen[i] == black_Queen[n] || abs(i - n) == abs(black_Queen[i] - black_Queen[n]))
            return false;
    }
    return true;
}
bool Check_White(int n)
{
    for(int i = 0;i < n;++i)
    {
        if(white_Queen[i] == white_Queen[n] || abs(i - n) == abs(white_Queen[i] - white_Queen[n]))
            return false;
    }
    return true;
}
void WhiteQueen(int n, int num)
{
    if(n >= num)
    {
        cot++;
        return;
    }
    for(int i = 0;i < num;++i)
    {
        if(i == black_Queen[n] || vis[n][i] == 0) continue;
        white_Queen[n] = i;
        if(Check_White(n))
        {
            WhiteQueen(n + 1,num);
        }
    }
}
void BlackQueen(int n, int num)
{
    if(n >= num)
    {
        WhiteQueen(0, num);
        return;
    }
    for(int i = 0;i < num;++i)
    {
        if(vis[n][i] == 0) continue;
        black_Queen[n] = i;
        if(Check_Black(n) == true)
        {
            BlackQueen(n + 1,num);
        }
    }
}
int main()
{
    cin >> num;
    for(int i = 0;i < num;++i)
    {
        for(int j = 0;j < num;++j)
        {
            scanf("%d",&vis[i][j]);
        }
    }
    BlackQueen(0,num);
    cout << cot << endl;
    return 0;
}

做法和n皇后类似，尽量不直接用二维数组来存放数据，用一维数组放皇后，下标值表示皇后在第几行，保存的值表示皇后在这一行的第几列。

涉及某些位置不能放皇后的问题，开一个数组vis进行标记，值为1表示可以放皇后，值为0表示放不了皇后。

依次放皇后，在每一行的所有列都进行尝试放上皇后，如果某个位置与之前放的所有皇后都不冲突，即截止目前放过的皇后都不在同一列和同一对角线（一维数组的情况下一行只会存在一个皇后无需判断），就表示这种情况成立可以继续往下一行进行搜索，递归调用即可。

先放黑皇后，再放白皇后。黑皇后递归函数如果判断到已经放完就可以递归调用白皇后函数。黑皇后放完之后占了一定的空间，在放白皇后时需要进行额外的判断，即判断该点是否已经有了黑皇后，如果有就要跳过当前位置搜索下一个位置。同样进行递归调用即可。