leetcode 面试题 08.12. 八皇后

原创已于 2024-06-22 17:45:01 修改 · 500 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode #算法 #N皇后 #c++

于 2024-06-22 13:15:14 首次发布

. - 力扣（LeetCode）

设计一种算法，打印 N 皇后在 N × N 棋盘上的各种摆法，其中每个皇后都不同行、不同列，也不在对角线上。这里的“对角线”指的是所有的对角线，不只是平分整个棋盘的那两条对角线。

注意：本题相对原题做了扩展

示例:

 输入：4
 输出：[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
 解释: 4 皇后问题存在如下两个不同的解法。
[
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]

class Solution {
public:
    bool isValid(int col, int row, int n, vector<int>& pos) {
        // 判断当前点是否可以时，只需要判断一下之前处理过的列。
        for (int c = 1; c < col; c++) {
            // 如果之前某一列同一行放置过，或者在对角线上
            if (pos[c] == row || col - c == abs(pos[c] - row)) {
                return false;
            }
        }
        return true;
    }

    void dfs(int col, int n, vector<int>& pos, vector<vector<string>>& res) {
        for (int r = 1; r <= n; r++) {
            if (isValid(col, r, n, pos)) {
                pos[col] = r;
                if (col == n) {
                    vector<string> all_str(n, string(n, '.'));
                    for (int cc = 1; cc <= n; cc++) {
                        all_str[pos[cc]-1][cc-1] = 'Q';
                    }
                    res.push_back(all_str);
                    continue;
                }

                dfs(col+1, n, pos, res);
                // 对于某一列col，针对这一行r，如果已经搜索成功，则需要置为0，这样可以再遍历其他合法的r
                pos[col] = 0;
            }
        }
    }

    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> res;
        // 存放每一列上皇后所在的行
        vector<int> pos(n+1, 0);
        // 对于第一个皇后处于第一列，看看有多少中摆法
        dfs(1, n, pos, res);
        return res;
    }
};