寒假Day58：蓝桥杯2n皇后问题八皇后变形

题目链接：

http://oj.ecustacm.cn/problem.php?id=1404

思路：

经典八皇后变形问题

标记行、判断列和两条对角线即可

先下黑皇后，下完后再去下白皇后

注意：这一题是1可以放，0不可以放

判断方式：

int check(int *queen,int x)//判断列和两条对角线
{
    for(int i=1; i<x; i++)
    {
        int w=queen[i]-queen[x];
        if(w==0||w==i-x||w==x-i)
            return 0;
    }
    return 1;
}

如何下黑皇后：

int black(int x)
{
    if(x==n+1)
        white(1);
    else
    {
        for(int i=1; i<=n; i++)
        {
            if(a[x][i]==1)
            {
                hei[x]=i;
                if(check(hei,x))
                    black(x+1);//递归下一行
            }
        }
    }
}

下白皇后的时候需要注意下不要和黑皇后重复了：

if(a[x][i]==1&&i!=hei[x])

AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

int a[10][10],hei[10],bai[10],ans,n;

int check(int *queen,int x)//判断列和两条对角线
{
    for(int i=1; i<x; i++)
    {
        int w=queen[i]-queen[x];
        if(w==0||w==i-x||w==x-i)
            return 0;
    }
    return 1;
}

void white(int x)
{
    if(x==n+1)
    {
        ans++;
        return;
    }
    for(int i=1; i<=n; i++)
    {
        if(a[x][i]==1&&i!=hei[x])
        {
            bai[x]=i;//在(x,i)放白皇后
            if(check(bai,x))
                white(x+1);
        }
    }
}

int black(int x)
{
    if(x==n+1)
        white(1);
    else
    {
        for(int i=1; i<=n; i++)
        {
            if(a[x][i]==1)
            {
                hei[x]=i;
                if(check(hei,x))
                    black(x+1);//递归下一行
            }
        }
    }
}

int main()
{
    cin>>n;
    ans=0;
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n; j++)
            cin>>a[i][j];
    }
    black(1);//第一行
    cout<<ans<<endl;
    return 0;
}