蓝桥杯--2N皇后问题

http://47.104.209.207/problem/old1069

先dfs黑皇后然后dfs白皇后，因为麻烦所以直接暴力check当前位置是否合法。

#include<iostream>
const int N=20;

using namespace std;
int s[N][N];
int n;
int res=0;
bool check(int r,int c,int queen){
    if(s[r][c]!=1){
        return false;
    }
    for(int i=0;i<n;i++){
        if(s[r][i]==queen||s[i][c]==queen){
            return false;
        }
    }
    int tr=r-min(r,c),tc=c-min(r,c);
    while(tr<n&&tc<n){
        if(s[tr][tc]==queen){
            return false;
        }
        tr++,tc++;
    }
    tr=r,tc=c;
    while(tr<n&&tc>=0) tr++,tc--;
    tr--,tc++;
    while(tr>=0&&tc<n){
        if(s[tr][tc]==queen){
            return false;
        }
        tr--,tc++;
    }
    return true;
}
void dfs_white(int cur){
    if(cur==n)
        res++;
    for(int i=0;i<n;i++){
        if(check(cur,i,3)){
            s[cur][i]=3;
            dfs_white(cur+1);
            s[cur][i]=1;
        }
    }
}
void dfs_black(int cur){
    if(cur==n)
        dfs_white(0);
    for(int i=0;i<n;i++){
        if(check(cur,i,2)){
            s[cur][i]=2;
            dfs_black(cur+1);
            s[cur][i]=1;
        }
    }
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            cin>>s[i][j];
        }
    }
    dfs_black(0);
    cout<<res<<endl;
    return 0;
}