P1219 [USACO1.5] 八皇后 Checker Challenge

#include <iostream>
#include <vector>
#include <algorithm>
#define M 13
using namespace std;
int l[2 * M + 5], r[2 * M + 5], ans = 0, cnt = 3;
vector<int> col;
void print_result(int k) {
	for (int i = 0; i < k; i++) {
		cout << col[i] << " ";
	}
	cout << endl;
	cnt--;
	return;
}
//i代表当前处理的是第i行
void fun(int i, int n) {
	if (i > n) {
		if (cnt) print_result(n);//输出前三种结果
		ans++;//记录结果总数
		return;
	}
	//j代表当前处理的是第j竖列
	for (int j = 1; j <= n; j++) {
		if (count(col.begin(), col.end(), j) 
			|| l[j + i - 1] 
			|| r[i - j + n]) continue;//这一竖行、左斜行或右斜行被占领过，就直接遍历下一竖行
		l[j + i - 1] = r[i - j + n] = 1;//宣布占领左斜行和右斜行
		col.push_back(j);//宣布占领竖行，存储到vector里是为了记忆占领顺序
		fun(i + 1, n);//递归下一行
		col.pop_back();//递归完了让位置
		l[j + i - 1] = r[i - j + n] = 0;//递归完了让位置
	}
	return;
}
int main() {
	int n;
	cin >> n;
	fun(1, n);
	cout << ans;
	return 0;
}