N-Queens n皇后问题

八皇后问题：规则是皇后不能处于对角线，切皇后不能挨着

递归回溯法 求八皇后问题解的个数
 1 #include <cstdio>
#include <vector>
using namespace std;
const int maxn = 1003;

int n,sum = 0;
vector<int> a;

bool isValid(int k,int i){
    bool flag =true;
    for(int kp=1;kp<a.size();kp++){
        if(a[kp]==i||abs(a[kp]-i)==abs(kp-k)){
            flag=false;
            break;
        }
    }
    return flag;
}

void queen(int k){
    if(k>n){
        sum++;
        a.pop_back();
        return;
    }
    for(int i=1;i<=n;i++){
        if(isValid(k,i)){
            a.push_back(i);
            queen(k+1);
        }
    }
    a.pop_back();
}

int main(){
    a.push_back(-1);
    printf("please input...\n");
    scanf("%d",&n);
    queen(1);
    printf("the number of the solution %d\n",sum);
}