[LeetCode] 51. N-Queens

最新推荐文章于 2024-07-12 20:18:19 发布

hiimdaosui

最新推荐文章于 2024-07-12 20:18:19 发布

阅读量274

点赞数

分类专栏： LeetCode 文章标签： leetcode dfs

本文链接：https://blog.csdn.net/hiimdaosui/article/details/51997179

版权

LeetCode 专栏收录该内容

184 篇文章 0 订阅

订阅专栏

思路:
DFS. 主要思路是从上向下一行一行添加Queen. 添加时查看该列上方是否有Queen, 该格子左上45度斜对角线和右上45度斜对角线有无Queen即可.

bool isValid(vector<string>& board, int row, int col) {
    // 判断此列上方有没有Queen
    for (int i = 0; i < row; i++) 
        if (board[i][col] == 'Q')
            return false;
    // 判断右上延伸对角线有没有Queen
    for (int i = row - 1, j = col + 1; i >= 0 && j < board[0].size(); i--, j++)
        if (board[i][j] == 'Q')
            return false;
    // 判断左上延伸对角线有没有Queen
    for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--)
        if (board[i][j] == 'Q')
            return false;

    return true;
}

void dfs(vector<vector<string>>& res, vector<string>& board, int row, int& n) {
    if (row == n)
        return res.push_back(board);
    for (int col = 0; col < n; col++) {
        if (isValid(board, row, col)) {
            board[row][col] = 'Q';
            dfs(res, board, row + 1, n);
            board[row][col] = '.';
        }
    }
}

vector<vector<string>> solveNQueens(int n) {
    vector<vector<string>> res;
    if (! n) return res;
    vector<string> board(n, string(n, '.'));
    dfs(res, board, 0, n);
    return res;
}