萌新刚学OI，求助一道站外题

题目描述 Description
有n个棋子被放置在n*n棋盘上，使得每行、每列有且只有一个，每条对角线(包括两条主对角线的所有平行线)上至多有一个棋子。
请编写程序找出所有棋子放置位置的解，解的输出按字典顺序排列。
请输出前3个解。最后一行是解的总个数。

输入描述 Input Description
一个数字N (6 <= N <= 13) 表示棋盘是N x N大小的。

输出描述 Output Description
前三行为前三个解，每行n个数字，每个数字分别表示皇后从第1行到第n行所对应的列数，数字之间用空格隔开。
第四行只有一个数字，表示解的总数。

样例输入 Sample Input
6
样例输出 Sample Output
2 4 6 1 3 5
3 6 2 5 1 4
4 1 5 2 6 3
4
数据范围及提示 Data Size & Hint
深度搜索

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
int a[22], b[22], c[22], d[22], cnt=0, n;
void dfs(int i) {	//给第i层放皇后 
    if(i==n+1) {
        cnt++;
        if(cnt<=3)
        {
        	for(int i=1; i<=n; i++) {
     	   		cout << a[i] <<' ';
   			}
   			cout << endl;
		}
        return;
    }
    for(int j=1; j<=n; j++) {
        if(!b[j]&&!c[i-j+n]&&!d[i+j]) {
            a[i] = j;
            b[j] = true;
            c[i-j+n] = true;
            d[i+j] = true;
            dfs(i+1);
            b[j] = false;
            c[i-j+n] = false;
            d[i+j] = false;
        } 
    }	
} 
int main() {
	cin >> n;
    dfs(1);
    cout << cnt;
    return 0;
}