[USACO1.5] 八皇后 Checker Challenge ( c++ )

题目描述

检查一个如下的6 x 6的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行，每列，每条对角线(包括两条主对角线的所有对角线)上都至多有一个棋子，如下例，就是一种正确的布局。

0   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 (6<=N<=14) 表示棋盘是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

题解思路

这是一个经典的N皇后问题，只不过这里的棋子是跳棋。我们需要在N x N的棋盘上放置N个棋子，使得每行、每列以及每条对角线上都只有一个棋子。

主要思路

1. 回溯法：我们可以使用回溯法来解决这个问题。从第一行开始，尝试在每一行的每一列放置一个棋子，并检查是否满足条件（即不与之前的棋子在同一列或同一对角线上）。
2. 对角线检查：为了快速检查对角线冲突，我们可以使用两个布尔数组来记录主对角线和副对角线上是否有棋子。主对角线的索引为 row - col，副对角线的索引为 row + col。
3. 字典序输出：由于回溯法会按字典序尝试每一列，因此找到的解自然按字典序排列。

具体步骤

1. 初始化一个布尔数组 col 用于记录每一列是否已经有棋子。
2. 初始化两个布尔数组 diag1 和 diag2 用于记录主对角线和副对角线上是否有棋子。
3. 使用递归函数 solve 尝试在每一行的每一列放置棋子，并递归到下一行。
4. 如果找到一个解，记录并继续寻找下一个解，直到找到所有解。
5. 输出前三个解和总解数。

#include <iostream>
#include <vector>
using namespace std;

int N;
vector<vector<int>> solutions;
vector<bool> col, diag1, diag2;

void solve(int row, vector<int>& current) {
    if (row == N) {
        solutions.push_back(current);
        return;
    }
    for (int c = 0; c < N; ++c) {
        if (!col[c] && !diag1[row - c + N - 1] && !diag2[row + c]) {
            col[c] = diag1[row - c + N - 1] = diag2[row + c] = true;
            current.push_back(c + 1);
            solve(row + 1, current);
            current.pop_back();
            col[c] = diag1[row - c + N - 1] = diag2[row + c] = false;
        }
    }
}

int main() {
    cin >> N;
    col.assign(N, false);
    diag1.assign(2 * N - 1, false);
    diag2.assign(2 * N - 1, false);

    vector<int> current;
    solve(0, current);

    for (int i = 0; i < min(3, (int)solutions.size()); ++i) {
        for (int j = 0; j < N; ++j) {
            if (j > 0) cout << " ";
            cout << solutions[i][j];
        }
        cout << endl;
    }
    cout << solutions.size() << endl;

    return 0;
}