基础练习 2n皇后问题

#include<iostream>
#include<math.h>

using namespace std;
#define Len 1000

int b[Len][Len];
int a[Len]= {0};
int c[Len]= {0};
int count=0;
int m;
bool check(int n, int i, int k) {
    if(b[n][i] == 0){
        return false;
    }
    if(k==0) {
        for(int j=0; j<n; j++) {
            if( a[j] == i || fabs(n-j) == fabs(a[j] - i)) {
                return false;
            }
        }
    } else {
        for(int j=0; j<n; j++) {
            if(c[j] == i || fabs(n-j) == fabs(c[j] - i)) {
                return false;
            }
        }
    }
    return true;
}


void f2(int n) {
    if(n==m) {
        count++;
        return;
    }

    for(int i=0; i<m; i++) {
        if(check(n, i, 1)) {
            c[n]=i;
            f2(n+1);
        }
    }
}

void f(int n) {
    if(n==m) {
        f2(0);

        //count++;
        return;
    }

    for(int i=0; i<m; i++) {
        bool x = check(n, i, 0);
        if(x) {
            a[n]=i;
            b[n][i]=0;
            f(n+1);
            b[n][i]=1;
        }
    }
}

int main() {

    cin>>m;
    for(int i=0; i<m; i++) {
        for(int j=0; j<m; j++) {
            cin>>b[i][j];
        }
    }
    f(0);
    cout<<count;
}