树的递归回溯 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;
}