Leetcode 51. N 皇后

最新推荐文章于 2024-11-01 15:19:21 发布

Bernard_Yang

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量2k

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_38224810/article/details/121895839

版权

LeetCode 专栏收录该内容

19 篇文章 0 订阅

订阅专栏

在这里插入图片描述

思路

class Solution:
    def solveNQueens(self, n: int) -> List[List[str]]:
        if not n: return []
        chessboard = [['.'] * n for _ in range(n)]
        res = []
        def isValid(row, col, chessboard):
            #检查列
            for i in range(len(chessboard)):#这是一个剪枝
                if chessboard[i][col] == 'Q':
                    return False    
            
            # 检查 45度角是否有皇后
            i = row -1
            j = col -1
            while i>=0 and j>=0:
                if chessboard[i][j] == 'Q':
                    return False
                i -= 1
                j -= 1
            # 检查 135度角是否有皇后
            i = row - 1
            j = col + 1
            while i>=0 and j < len(chessboard):
                if chessboard[i][j] == 'Q':
                    return False
                i -= 1
                j += 1
            return True
        
        def backtracking(board, row, n):
            # 如果走到最后一行，说明已经找到一个解
            if row == n:
                temp_res = []
                for temp in board:
                    temp_str = "".join(temp)
                    temp_res.append(temp_str)
                res.append(temp_res)
                return
            for col in range(n):
                if not isValid(row, col, board):
                    continue
                board[row][col] = 'Q'
                backtracking(board, row+1, n)
                board[row][col] = '.'
        backtracking(chessboard, 0, n)
        return res