二叉树的层次遍历- python-队列

题目:

思路:

层次遍历要求从左到右访问节点,符合先进先出的顺序,所以要用到队列,在python中一般用deque双向队列

  1. 判断root是否为空,若为空,返回空列表
  2. 把root加入到队列中
  3. 当队列不为空
    1. for _ in range(len(queue)):
      1. 过渡列表temp=[]
      2. 取出队列的头部节点,并加入到temp中
      3. 判断该节点的左右子节点是否为空,不为空加入到队列中
    2. 一层的元素全部存在temp中,把temp加入到最终答案ans中
  4. 输出ans

代码:

复制代码
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def levelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
        if not root:
            return []
        res, queue = [], deque()
        queue.append(root)
        while queue:
            temp = []
            for _ in range(len(queue)):
                node = queue.popleft()
                temp.append(node.val)
                if node.left:queue.append(node.left)
                if node.right:queue.append(node.right)
            res.append(temp)
        return res
相关推荐
笨笨阿库娅7 小时前
从零开始的算法基础学习
学习·算法
不想睡觉_7 小时前
优先队列priority_queue
c++·算法
那个村的李富贵15 小时前
CANN加速下的AIGC“即时翻译”:AI语音克隆与实时变声实战
人工智能·算法·aigc·cann
power 雀儿15 小时前
Scaled Dot-Product Attention 分数计算 C++
算法
琹箐16 小时前
最大堆和最小堆 实现思路
java·开发语言·算法
renhongxia116 小时前
如何基于知识图谱进行故障原因、事故原因推理,需要用到哪些算法
人工智能·深度学习·算法·机器学习·自然语言处理·transformer·知识图谱
坚持就完事了16 小时前
数据结构之树(Java实现)
java·算法
算法备案代理17 小时前
大模型备案与算法备案,企业该如何选择?
人工智能·算法·大模型·算法备案
赛姐在努力.17 小时前
【拓扑排序】-- 算法原理讲解,及实现拓扑排序,附赠热门例题
java·算法·图论
野犬寒鸦18 小时前
从零起步学习并发编程 || 第六章:ReentrantLock与synchronized 的辨析及运用
java·服务器·数据库·后端·学习·算法