Leetcode—51.N皇后【困难】-Toy模板网

这篇具有很好参考价值的文章主要介绍了Leetcode—51.N皇后【困难】。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

2023每日刷题（八十四）

Leetcode—51.N皇后

Leetcode—51.N皇后【困难】,LeetCode刷题,leetcode,算法,职场和发展,c++,经验分享,dfs,回溯

算法思想

Leetcode—51.N皇后【困难】,LeetCode刷题,leetcode,算法,职场和发展,c++,经验分享,dfs,回溯

实现代码

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<int> col(n), on_path(n), diag1(2*n), diag2(2*n);
        vector<vector<string>> ans;
        function<void(int)> dfs = [&](int r) {
            if(r == n) {
                vector<string> res(n);
                for(int i = 0; i < n; i++) {
                    res[i] = string(col[i], '.') + 'Q' + string(n - 1 - col[i], '.');
                }
                ans.emplace_back(res);
                return;
            }
            for(int c = 0; c < n; c++) {
                int rc = r - c + n - 1;
                if(!on_path[c] && !diag1[r + c] && !diag2[rc]) {
                    col[r] = c;
                    on_path[c] = diag1[r + c] = diag2[rc] = 1;
                    dfs(r + 1);
                    on_path[c] = diag1[r + c] = diag2[rc] = 0;
                }
            }
        };
        dfs(0);
        return ans;
    }
};