蓝桥杯基础练习VIP-2n皇后问题(N皇后问题之上再嵌套一个n皇后就行！)

#include<iostream>

using namespace std;

int a[100][100],n,ans=0;

bool vis1[100],vis2[100];

bool x1[100],y1[100],x2[100],y2[100];

void dfs(int r){

    if(r==n+1){

        ans++;

        return ;

    }

    for(int i=1;i<=n;i++){

        if(!vis1[i]&&a[r][i]&&!x1[r+i]&&!y1[r-i+n]){

            vis1[i]=x1[r+i]=y1[r-i+n]=true,a[r][i]=0;

            for(int j=1;j<=n;j++){

                if(!vis2[j]&&a[r][j]&&!x2[r+j]&&!y2[r-j+n]){

                    vis2[j]=x2[r+j]=y2[r-j+n]=true,a[r][j]=0;

                    dfs(r+1);

                    vis2[j]=x2[r+j]=y2[r-j+n]=false,a[r][j]=1;

                }

            }

            vis1[i]=x1[r+i]=y1[r-i+n]=false,a[r][i]=1;

        }

    }

}

int main()

{

    cin>>n;

    for(int i=1;i<=n;i++){

        for(int j=1;j<=n;j++){

            cin>>a[i][j];

        }

    }

    dfs(1);

    cout<<ans<<endl;

    return 0;

}