LeetCode102题：二叉树的层序遍历（python3）

Zachos

已于 2024-03-12 16:21:05 修改

阅读量479

点赞数 13

分类专栏：银河骑士每日一练文章标签： python 开发语言数据结构算法

于 2024-03-12 16:01:59 首次发布

本文链接：https://blog.youkuaiyun.com/Zachos/article/details/136654973

版权

银河骑士每日一练专栏收录该内容

59 篇文章

订阅专栏

文章介绍了如何使用Python的collections库中的deque数据结构进行树的层次遍历，利用deque的popleft方法实现O(1)时间复杂度，解决在队列头部操作的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
代码思路：使用队列先进先出的特性，queue[]不为空进入for循环，tmp存储每层的节点，将结果添加至res[]中。
python中使用collections中的双端队列deque()，其popleft()方法可达到O(1)时间复杂度。

class Solution:
    def levelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
        if not root:return[]
        res,queue = [],collections.deque()
        queue.append(root)
        while queue:
            tmp = []
            for _ in range(len(queue)): #len(queue)的长度不为null就进入循环
                node = queue.popleft()
                tmp.append(node.val)
                if node.left:queue.append(node.left)
                if node.right:queue.append(node.right)  
            res.append(tmp)
        return res

列表的方法一般只支持尾部，例如pop、append，但是对于一些队列问题，需要在头部操作，则需要使用pop(0)、insert(0,x)等方法，不太方便，Python的collections包中的双端队列deque很好地解决了这个问题，增加了头部方法popleft、appendleft等。

from collections import deque

alist=['a','b','c','d']
dq=deque(alist)
print(dq.pop())
print(dq.popleft())
print(dq)
dq.appendleft('e')
print(dq)

d
a
deque(['b', 'c'])
deque(['e', 'b', 'c'])