LeetCode题解(0051):N皇后问题(Python)
发布日期:2021-06-29 19:58:55 浏览次数:2 分类:技术文章

本文共 1418 字,大约阅读时间需要 4 分钟。

题目:(困难)

标签:回溯算法

解法 时间复杂度 空间复杂度 执行用时
Ans 1 (Python) O ( N ! ) O(N!) O(N!) O ( N ) O(N) O(N) 48ms (69.07%)
Ans 2 (Python)
Ans 3 (Python)

解法一(回溯算法):

class Solution:    def solveNQueens(self, n: int) -> List[List[str]]:        ans = []        now = []  # 当前每行中皇后的位置        columns = set()  # 当前已有皇后的列        diagonal_right = set()  # 当前已有皇后的从右上到左下的斜线        diagonal_left = set()  # 当前已有皇后的从左上到右下的斜线        def track_back(row):            # 处理已经回溯完成的情况            if row == n:                ans.append(["".join(["Q" if now[i] == j else "." for j in range(n)]) for i in range(n)])            # 处理还没有回溯完成的情况            else:                for j in range(n):  # 遍历当前行中所有的列                    if j in columns:  # 判断当前列是否已被占用                        continue                    if j + row in diagonal_right:  # 判断当前从右上到左下的斜线是否已被占用                        continue                    if j - row in diagonal_left:  # 判断当前从左上到右下的斜线是否已被占用                        continue                    # 继续进入当前分支                    now.append(j)                    columns.add(j)                    diagonal_right.add(j + row)                    diagonal_left.add(j - row)                    track_back(row + 1)                    # 移除当前分支                    now.pop()                    columns.remove(j)                    diagonal_right.remove(j + row)                    diagonal_left.remove(j - row)        track_back(0)        return ans

转载地址:https://dataartist.blog.csdn.net/article/details/108386397 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:LeetCode题解(0060):计算1到n的按大小顺序的第k个排列(Python)
下一篇:LeetCode题解(Offer20):识别表示数值的字符串(支持正负号、E)(Python)

发表评论

最新留言

做的很好,不错不错
[***.243.131.199]2024年04月25日 04时20分27秒