LeetCode | N-Queens I,II

最新推荐文章于 2024-09-16 06:54:08 发布

益达915

最新推荐文章于 2024-09-16 06:54:08 发布

阅读量241

点赞数

分类专栏： LeetCode 算法 DFS 文章标签： leetcode 算法 DFS

本文链接：https://blog.csdn.net/u013033845/article/details/52222954

版权

算法同时被 3 个专栏收录

114 篇文章 0 订阅

订阅专栏

LeetCode

100 篇文章 0 订阅

订阅专栏

DFS

9 篇文章 0 订阅

订阅专栏

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.
N后棋盘
Given an integer n, return all distinct solutions to the n-queens puzzle.

Each solution contains a distinct board configuration of the n-queens’ placement, where ‘Q’ and ‘.’ both indicate a queen and an empty space respectively.

For example,
There exist two distinct solutions to the 4-queens puzzle:

[
[“.Q..”, // Solution 1
“…Q”,
“Q…”,
“..Q.”],

[“..Q.”, // Solution 2
“Q…”,
“…Q”,
“.Q..”]
]

经典的N后问题，深搜
这里面有几个问题的处理条件需要注意：
1、如何记录中间结果？
一开始我所想是用一个vector< string> &temp,来维护一个类似于二维数组，每次对step行填充皇后，并保证noConflict。
也即

if(noConflict()){
    temp[i][j]='Q';
    dfs(....
    temp[i][j]='.';
}

不过看了参考答案发现他是用一个vector< int> &temp这个一维数组来完成的，temp[i]表示在i行的temp[i]位置放置皇后。
这种的辅助空间自然也就小了很多，等n行填满之后，只需要将整个棋盘复原出来
2、判断对角线冲突
abs(i-x)==abs(temp[i]-y)
即x1-x2,y1-y2的绝对值不相等。

class Solution {
public:
    int n;
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string> > result;
        vector<int> temp(n,-1);//表示第i行皇后所在的位置编号
        this->n=n;
        dfs(temp,0,result);
        return result;
    }

    void dfs(vector<int> &temp,int step,vector<vector<string> > &result){
        if(step==n){
            //将temp转化为字符串
            vector<string> answer;
            for(int i=0;i<n;i++) {
                string s(n,'.');
                for(int j=0;j<s.size();j++){
                    if(temp[i]==j) s[j]='Q';
                }
                answer.push_back(s);
            }
            result.push_back(answer);
            return;
        }

        //尝试在i处放棋子
        for(int i=0;i<n;i++){
            //在这个位置放没有冲突 
            if(noConflict(step,i,temp)){
                temp[step]=i;
                dfs(temp,step+1,result);
                temp[step]=-1;
            }
        }
    }

    //检测在x,y处放皇后不会冲突
    bool noConflict(int x,int y,vector<int> &temp){
        for(int i=0;i<x;i++){
            //在同一列 
            if(temp[i]==y) return false;
            //在同一对角线
            if(abs(i-x)==abs(temp[i]-y)) return false;
        }
        return true;
    }
};