LeetCode刷题笔记（N-Queens）

最近北京甚是凉爽，动不动就洒下雨点，刷新了我对北京不怎么下雨的观点。刚刷了一道被标为“hard”的题目，但实际上并不是很难，下面就和大家分享一下经验吧！

题目如下：

题意分析：

本题为著名的N皇后问题：要求将n个皇后摆放在n*n的棋盘格中，使得横、纵以及两个角度各自的2*n-1条对角线方向均不会同时出现两个皇后。请找到n皇后问题的所有解，解中要求皇后的位置用'Q'表示，非皇后的位置用 '.'表示。

方法一（递归回溯法①）

采取“逐行试凑法”，先用index记录试放皇后当前行，row记录n皇后问题的一个解，用布尔型数组col、diam1、diam2分别标记是否在纵向和两个角度对角线方向放过皇后，其中两个角度的对角线个数均为2*n-1，且右上到左下的对角线满足横纵坐标相加之和为定值，左上到右下的对角线满足横坐标-纵坐标为定值。然后创建findsolveNQueens函数进行递归调用，其中递归终止条件为“if ( index == n ) res.push_back( getResult( n, row) );”，当访问位置的纵向与两个角度对角线方向均未放过皇后，那么可以在该位置放下皇后且继续递归调用。由于这里需要回溯，所以需要清除之前的标记。最后，通过getResult函数将n皇后问题的每个解row转换成题目要求的解形式，并返回n皇后问题的所有解即可。

解题代码如下：

class Solution{
private:
    void findsolveNQueens(int n, int index, vector<int>& row, vector<bool>& col, vector<bool>& diam1, vector<bool>& diam2, vector<vector<string>>& res){
        if ( index == n ) res.push_back( getResult( n, row) );

        for (int i = 0; i < n; i++) {
            //由于数组diam2是从0开始访问的,所以整体需要+n-1,即index-i+n-1
            if( !col[i] && !diam1[index+i] && !diam2[index-i+n-1] )
                {
                    row.push_back(i);
                    col[i] = true;
                    diam1[index+i] = true;
                    diam2[index-i+n-1] = true;
                    findsolveNQueens(n, index+1, row, col, diam1, diam2, res);
                    col[i] = false;
                    diam1[index+i] = false;
                    diam2[index-i+n-1] = false;
                    row.pop_back();
             }
        }
    }

    vector<string> getResult( int n, vector<int>& row ){
        vector<string> result ( n, string(n,'.') );
        for ( int i = 0; i < n; i++ ) {
            result[i][row[i]] = 'Q';
        }
      return result;
    }


public:
    vector<vector<string>> solveNQueens( int n ){
        vector<vector<string>> res;
        vector<int> row;
        vector<bool> col( n, false );
        vector<bool> diam1( 2*n-1, false );
        vector<bool> diam2( 2*n-1, false );
        findsolveNQueens( n, 0, row, col, diam1, diam2 , res);

        return res;
    }
};

提交后的结果如下：

 

方法二（递归回溯法②）

与方法一不同在于：1、采用一个函数isValid去判断某个位置是否可以放皇后，而不再用布尔型数组去标记了；2、直接用获得问题的解，不用通过getResult函数将n皇后问题的每个解row转换成题目要求的解形式。

解题代码如下：

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> res;
        vector<string> queens(n, string(n, '.'));
        findsolveNQueens(0, queens, res);
        return res;
    }
    void findsolveNQueens(int index, vector<string>& queens, vector<vector<string>>& res) {
        int n = queens.size();
        if (index == n) {
            res.push_back(queens);
            return;
        }
        for (int i = 0; i < n; ++i) {
            if (isValid(queens, index, i)) {
                queens[index][i] = 'Q';
                findsolveNQueens(index + 1, queens, res);
                queens[index][i] = '.';
            }
        }
    }
    bool isValid(vector<string>& queens, int row, int col) {
        for (int i = 0; i < row; ++i) {
            if (queens[i][col] == 'Q') return false;
        }
        for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; --i, --j) {
            if (queens[i][j] == 'Q') return false;
        }
        for (int i = row - 1, j = col + 1; i >= 0 && j < queens.size(); --i, ++j) {
            if (queens[i][j] == 'Q') return false;
        }
        return true;
    }
};

提交后的结果如下：

 

 

 

日积月累，与君共进，增增小结，未完待续。