树的递归回溯 n皇后问题

#include <iostream>
#include <memory.h>
using namespace std;
int row[20];
int k,n;
int total=0;

void trial(int k){
int i=0;
if(k>=n){
total++;
for(int i=0;i<n;i++)
cout<<i<<","<<row[i]<<endl;
cout<<endl;
}
else{
for(;i<n;i++){
int flag=1;
for(int j=0;j<k;j++){  //check the columns before
if(i==row[j] ||   //同一列
i-k==row[j]-j || //对角线
k+i==row[j]+j  //反对角线
){
flag=0;  //不符合条件
break;
}
}//for
if(flag){  //符合条件
row[k]=i;
trial(k+1);
}
}//for
}
}

int main(){
cin>>n;
memset(row,0,sizeof(row));
trial(0);
cout<<total<<endl;
return 0;
}