C++ | Leetcode C++题解之第52题N皇后II

Ddddddd_158

于 2024-04-26 09:17:35 发布

阅读量471

点赞数 20

分类专栏：经验分享文章标签： C++ Leetcode 题解

本文链接：https://blog.csdn.net/Ddddddd_158/article/details/138205730

版权

经验分享专栏收录该内容

413 篇文章 4 订阅

订阅专栏

本文介绍了一种使用C++编写的解决方案，通过递归和回溯法解决N皇后问题，避免在同一行、对角线上放置两个皇后。函数totalNQueens和backtrack详细描述了算法过程。

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

题目：

题解：

class Solution {
public:
    int totalNQueens(int n) {
        unordered_set<int> columns, diagonals1, diagonals2;
        return backtrack(n, 0, columns, diagonals1, diagonals2);
    }

    int backtrack(int n, int row, unordered_set<int>& columns, unordered_set<int>& diagonals1, unordered_set<int>& diagonals2) {
        if (row == n) {
            return 1;
        } else {
            int count = 0;
            for (int i = 0; i < n; i++) {
                if (columns.find(i) != columns.end()) {
                    continue;
                }
                int diagonal1 = row - i;
                if (diagonals1.find(diagonal1) != diagonals1.end()) {
                    continue;
                }
                int diagonal2 = row + i;
                if (diagonals2.find(diagonal2) != diagonals2.end()) {
                    continue;
                }
                columns.insert(i);
                diagonals1.insert(diagonal1);
                diagonals2.insert(diagonal2);
                count += backtrack(n, row + 1, columns, diagonals1, diagonals2);
                columns.erase(i);
                diagonals1.erase(diagonal1);
                diagonals2.erase(diagonal2);
            }
            return count;
        }
    }
};