LeetCode 51. N皇后

题目描述

n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

上图为 8 皇后问题的一种解法。

给定一个整数 n，返回所有不同的 n 皇后问题的解决方案。

每一种解法包含一个明确的 n 皇后问题的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。

示例:

输入: 4
输出: [
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]
解释: 4 皇后问题存在两个不同的解法。

问题分析

此题为DFS+回溯的经典问题。有两个可变的递归参数：vec和row，vec是int型的数组，里面保存着当前各行已经保存的皇后的位置，初始值都设为-1。row表示要开始遍历第row行。每行从第0列开始到最后一列，依次判断能否放置于该列的位置上。用isvalid()函数判断。第0行任何一个位置都可以放置，没有限制，从第1行开始需要参考下面的各行的信息判断此列能否放置。如果判断某列可以放置皇后，那么将vec数组中的此行的某列记录上，然后进行下一次DFS。在DFS函数返回后，因为接下来要遍历到该行的下一列的元素判断能否放置，所以这时需要将该行的状态恢复。递归基是判断如果row等于vec.size()，说明此时vec中所有行都有皇后放置了，这时将创建出的对应的string类型的数组加到ans里面去并返回。最终ans里面保存的就是所有可能的情况了。

代码实现

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> ans;
        vector<int> vec(n, -1);
        DFS(vec, 0, ans);
        return ans;
    }
    
    void DFS(vector<int>& vec, int row, vector<vector<string>>& ans){
        if(row == vec.size()){
            vector<string> out(vec.size(), string(vec.size(), '.'));
            for(int i = 0; i < row; i++)
                out[i][vec[i]] = 'Q';
            ans.push_back(out);
            return;
        }   
        for(int i = 0; i < vec.size(); i++){
            if(row == 0 || isvalid(vec, row, i)){
                vec[row] = i;
                DFS(vec, row + 1, ans);
                vec[row] = -1;
            }    
        }
        return;
    }
    
    bool isvalid(vector<int>& vec, int row, int col){
        for(int i = 0; i < row; i++){
            if(col == vec[i] || row - i == col - vec[i] || i - row == col - vec[i])
                return false;
        }
        return true; 
    }
};