随机化算法的初步实践-N 后等问题

问题描述：

        N 皇后问题是一道经典的计算机科学问题，其目标是将 N 个皇后放置在 N ×N 的棋盘上，使得任意两个皇后都不能在同一行、同一列或同一对角线上。N 皇后问题 是一个 NP 完全问题，其解法通常需要使用到搜索算法。本实验采用拉斯维加斯算法。

算法设计：

        基本思路： 拉斯维加斯算法通过随机放置皇后来探索解空间，直到找到一个可行 解或达到最大迭代次数。

        策略选择： 在每次迭代中，随机放置皇后的位置是关键。为了确保皇后不在同一 行或同一列，可以使用一个大小为 N 的向量 pos 来记录每个皇后所在的列。在每次 迭代中，为 pos[i] 随机选择一个合法的列位置，即不与之前放置的皇后冲突。

        数据结构选择： 为了实现算法，可以选择以下数据结构：

        pos: 一个大小为 N 的整数向量，用于记录每个皇后所在的列位置。

        IsConflict(): 用于判断某个位置是否与之前放置的皇后冲突。

伪代码： 

function LasVegasNQueens(N):
    pos := vector of size N, initialized with -1
    
    iterations := 0
    
    while iterations < maxIterations:
        generateRandomSolution(pos)
        
        if isValidSolution(pos):
            return pos
        
        iterations := iterations + 1
    
    return empty vector

function generateRandomSolution(pos):
    for i := 0 to N-1:
        pos[i] := randomValidColumn()

function randomValidColumn():
    columns := list of valid column indices
    
    return random element from columns

function isValidSolution(pos):
    for i := 0 to N-2:
        for j := i+1 to N-1:
            if pos[i] == pos[j] or abs(i - j) == abs(pos[i] - pos[j]):
                return false
    
    return true

时间复杂度：O(N^2)，空间复杂性：O(N)

代码：

#include <iostream>
#include <vector>

using namespace std;

bool isConflict(const vector<int> &pos, int row, int col)
{
    for (int i = 0; i < row; ++i)
    {
        if (pos[i] == col || abs(i - row) == abs(pos[i] - col))
        {
            return true;
        }
    }
    return false;
}

bool validateSolution(const vector<int> &pos)
{
    int n = pos.size();
    for (int i = 0; i < n; ++i)
    {
        for (int j = i + 1; j < n; ++j)
        {
            if (pos[i] == pos[j] || abs(i - j) == abs(pos[i] - pos[j]))
            {
                return false;
            }
        }
    }
    return true;
}

void printBoard(const vector<int> &pos)
{
    int n = pos.size();
    for (int i = 0; i < n; ++i)
    {
        for (int j = 0; j < n; ++j)
        {
            if (pos[i] == j)
            {
                cout << "Q ";
            }
            else
            {
                cout << ". ";
            }
        }
        cout << endl;
    }
}

int main()
{
    vector<int> solution(8);

    cout << "请输入 8 个数字（范围：0 到 7），表示每个皇后的位置：" << endl;
    for (int i = 0; i < 8; ++i)
    {
        cin >> solution[i];
    }

    cout << "输入的解：" << endl;
    printBoard(solution);

    cout << "验证解的正确性：" << endl;
    if (validateSolution(solution))
    {
        cout << "解是正确的" << endl;
    }
    else
    {
        cout << "解是错误的" << endl;
    }
    system("pause");
    return 0;
}

测试用例：