[USACO 1.5.4]跳棋的挑战题解

[USACO 1.5.4]跳棋的挑战
这是一道比较简单n皇后题目哈
题目描述
一个如下的6 x 6的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行，每列，每条对角线(包括两条主对角线的所有对角线)上都至多有一个棋子。

列号 1 2 3 4 5 6-------------------------1 | | O | | | | |-------------------------2 | | | | O | | |-------------------------3 | | | | | | O |-------------------------4 | O | | | | | |-------------------------5 | | | O | | | |-------------------------6 | | | | | O | |-------------------------

上面的布局可以用序列2 4 6 1 3 5来描述，第i个数字表示在第i行的相应位置有一个棋子，如下： 行号 1 2 3 4 5 6 列号 2 4 6 1 3 5 这只是跳棋放置的一个解。
请遍一个程序找出所有跳棋放置的解。并把它们以上面的序列方法输出。解按字典顺序排列。请输出前3个解。最后一行是解的总个数。
特别注意: 对于更大的N(棋盘大小N x N)你的程序应当改进得更有效。不要事先计算出所有解然后只输出，这是作弊。如果你坚持作弊，一旦你被无聊的管理员LJ发现的话，我想后果会非常严重^ ^
输入一个数字N (6<= N <= 13) 表示棋盘是N x N大小的。
输出前三行为前三个解，每个解的两个数字之间用一个空格隔开。第四行只有一个数字，表示解的总数。
样例输入输出
输入#1
6
输出#1
2 4 6 1 3 5
3 6 2 5 1 4
4 1 5 2 6 3
4

#include <iostream>
#include <cstring>
using namespace std;
const int OFFSET = 15;  // 因为n<=8，15足够
int n, a[30], b[30], o[30], I[30], ans;
void dfs(int x) {if (x > n) {ans++;if (ans <= 3) {  // 只输出前三个解for (int i = 1; i <= n; ++i) {cout << a[i];if (i < n) cout << " ";}cout << endl;}return;}for (int i = 1; i <= n; ++i) {if (b[i] == 0 && o[i + x] == 0 && I[i - x + OFFSET] == 0) {o[i + x] = I[i - x + OFFSET] = b[i] = 1;a[x] = i;dfs(x + 1);o[i + x] = I[i - x + OFFSET] = b[i] = 0;}}
}
int main() {cin >> n;// 初始化数组memset(b, 0, sizeof(b));memset(o, 0, sizeof(o));memset(I, 0, sizeof(I));dfs(1);cout << ans << endl;return 0;
}