LeetCode-51-N 皇后

最新推荐文章于 2024-09-22 04:59:47 发布

犬兄的海角

最新推荐文章于 2024-09-22 04:59:47 发布

阅读量216

点赞数

分类专栏： LeetCode-待复习 LeetCode刷题记录文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_43825194/article/details/129931251

版权

LeetCode刷题记录同时被 2 个专栏收录

312 篇文章 1 订阅

订阅专栏

LeetCode-待复习

145 篇文章 1 订阅

订阅专栏

文章介绍了如何使用回溯法来解决N皇后问题，通过维护列、主对角线和副对角线的状态，避免皇后之间的冲突。在每行中寻找可放置皇后的安全位置，如果找到所有行的解，则将当前解决方案保存。递归过程中，会恢复当前节点的状态以尝试其他可能性。

摘要由CSDN通过智能技术生成

在这里插入图片描述

1、回溯

我们可以利用回溯搜索所有的可能。我们利用数组column记录每一列的使用情况，数组ldiag(2 * n - 1, false)记录主对角线上的使用情况，数组rdiag(2 * n - 1, false)记录副对角线上的使用情况。值得注意的是，我们主对角线上的序号是自左下到右上，副对角线上的序号是自左上到右下。

在回溯当中，我们按照每行寻找有无可以放置的位置，若此时行数等于总行数则直接将当前结果压入最终结果中。否则我们首先判断当前列、主对角线和副对角线上的所在列是否有人占用，有则直接跳过，否则我们将当前节点位置修改为Q并递归调用函数。递归结束后我们将当前节点的状态改回为之前的状态并修改矩阵的值。

class Solution {
public:
    // 辅函数
    void backtracking(vector<vector<string>> &ans, vector<string> &board, vector<bool> &column, vector<bool> &ldiag, vector<bool> &rdiag, int row, int n) {
        if (row == n) {
            ans.push_back(board);
            return;
        }
        for (int i = 0; i < n; ++i) {
            if (column[i] || ldiag[n - row + i - 1] || rdiag[row + i]) {
                continue;
            }
            // 修改当前节点状态
            board[row][i] = 'Q';
            column[i] = ldiag[n - row + i - 1] = rdiag[row + i] = true;
            // 递归子节点
            backtracking(ans, board, column, ldiag, rdiag, row + 1, n);
            // 回改当前节点状态
            board[row][i] = '.';
            column[i] = ldiag[n - row + i - 1] = rdiag[row + i] = false;
        }
    }

    // 主函数
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> ans;
        if (n == 0) {
            return ans;
        }
        vector<string> board(n, string(n, '.'));
        vector<bool> column(n, false), ldiag(2 * n - 1, false), rdiag(2 * n - 1, false);
        backtracking(ans, board, column, ldiag, rdiag, 0, n);
        return ans;
    }
};