Leetcode101_回溯法_51N皇后问题

诶你爱我吗

已于 2022-07-18 19:20:39 修改

阅读量280

点赞数

分类专栏： LeetCode 文章标签：算法深度优先 leetcode

于 2022-07-18 19:19:15 首次发布

本文链接：https://blog.csdn.net/weixin_44244154/article/details/125858341

版权

LeetCode 专栏收录该内容

60 篇文章 1 订阅

订阅专栏

N皇后问题

在这里插入图片描述

题解

参考资料讲解讲的很清晰
a[i]表示在第i - 1行的皇后在第a[i] - 1列
遍历每一列，确定每一行皇后的位置并回溯
注意判断是否冲突的条件即皇后是否能放在第i行第j列，要看每一列，左斜和右斜是否冲突

class Solution {
public:
    bool check(int x, int y, vector<string>& king,vector<int>& a){
        for(int i = 0; i < x; i++){
            if(king[i][y] == 'Q') return false;
            if(i + a[i] == x + y) return false;
            if(i - a[i] == x - y) return false;
        }
        return true;
    }
    void dfs(int row,int n, vector<string>& king,vector<int>& a, vector<vector<string>>& ans){
        if(row == n){
            ans.push_back(king);
            return;
        }
        for(int i = 0; i < n; i++){            
            if(check(row,i,king,a)){
                a[row] = i;
                king[row][i] = 'Q';
                dfs(row + 1,n,king,a,ans);
                king[row][i] = ('.');
                a[row] = 0;
            }
        }
    }
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> ans;
        vector<int> a(n);
        vector<string> king(n,string(n,'.'));
        dfs(0,n,king,a,ans);
        return ans;
    }
};