Leetcode 51. N 皇后

JehanRio

已于 2023-09-21 09:44:54 修改

阅读量126

点赞数

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

于 2023-05-23 10:35:50 首次发布

本文链接：https://blog.csdn.net/weixin_51322383/article/details/130821576

版权

Leetcode专栏专栏收录该内容

278 篇文章 0 订阅

订阅专栏

文章目录

题目
代码（首刷部分看解析）
代码（8.24 部分看解析）
代码（9.21 三刷自解）

题目

在这里插入图片描述
Leetcode 51. N 皇后

代码（首刷部分看解析）

class Solution {
public:
    vector<vector<string>> res;
    vector<string> board;
    bool check(int n , int x, int y) {
        for(int i = 0; i < x; i++) {    // 同一列
            if(board[i][y] == 'Q')
                return false;
        }
        for(int i = x-1, j = y+1; i >= 0 && j < n; i--,j++) {
            if(board[i][j] == 'Q')
                return false;
        }
        for(int i = x-1, j = y-1; i >= 0 && j >= 0; i--,j--) {
            if(board[i][j] == 'Q')
                return false;
        }
        return true;
    }   
    vector<vector<string>> solveNQueens(int n) {
        board = move(vector<string>(n, string(n,'.')));
        backtrack(n, 0);
        return res;
    }
    void backtrack(int n, int x) {  // x表示第几行
        if(x == n) {
            res.push_back(board);
            return;
        }
        for(int i = 0; i < n; i++) {
            if(!check(n, x, i)) {
                continue;
            }
            board[x][i] = 'Q';
            backtrack(n, x+1);
            board[x][i] = '.';
        }
    }
};

代码（8.24 部分看解析）

class Solution {
public:
    int res;
    int Nqueen(int n) {
        res = 0;
        vector<vector<bool>> grid(n, vector<bool>(n, false));
        backtrack(grid, n, 0);
        return res;
    }
    void backtrack(vector<vector<bool>> grid, int n, int x) {
        if(x == n) {
            res++;
            return;
        }
        for(int i = 0; i < n; i++) {
            if(check(grid, x, i, n)) {
                grid[x][i] = true;
                backtrack(grid, n, x+1);
                grid[x][i] = false;
            }
        }
    }
    bool check(vector<vector<bool>> grid, int x, int y, int n) {
        for(int i = 0; i < x; i++) {
            if(grid[i][y])
                return false;
        }
        for(int i = x - 1, j = y - 1; i >= 0 && j >= 0; i--, j--) {
            if(grid[i][j])
                return false;
        }
        for(int i = x - 1, j = y + 1; i >= 0 && j < n; i--, j++) {
            if(grid[i][j])
                return false;
        }
        return true;
    }
};

代码（9.21 三刷自解）

class Solution {
public:
    vector<vector<string>> res;
    vector<string> map;
    vector<vector<string>> solveNQueens(int n) {
        map = vector<string>(n, string(n, '.'));
        backtrack(map, n, 0);
        return res;
    }
    void backtrack(vector<string>& map, int n, int x) {
        if(x == n) {
            res.emplace_back(map);
            return;
        }
        for(int i = 0; i < n; i++) {
            if(!check(x, i, n)) continue;
            map[x][i] = 'Q';
            backtrack(map, n, x+1);
            map[x][i] = '.';
        }
    }
    bool check(int x, int y, int n) {
        for(int i = 0; i < x; i++) {
            if(map[i][y] == 'Q')
                return false;
        }
        for(int i = 0; i < y; i++) {
            if(map[x][i] == 'Q')
                return false;
        }
        for(int i = x-1, j = y-1; i >= 0 && j >= 0; i--, j--) {
            if(map[i][j] == 'Q')
                return false;
        }
        for(int i = x-1, j = y+1; i >= 0 && j < n; i--, j++) {
            if(map[i][j] == 'Q')
                return false;
        }
        return true;
    }
};