102. 二叉树的层序遍历

最新推荐文章于 2024-09-14 18:52:56 发布

*靖

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量169

点赞数 4

文章标签： leetcode 广度优先

本文链接：https://blog.csdn.net/weixin_43280025/article/details/141613539

版权

102. 二叉树的层序遍历

思路

1.题中的第一个案例，如果按照层序遍历，会是 3 9 20 15 7，那就要想办法按照这个顺序放入，再按照这个顺序取出，即先进先出，想到了队列
2. 用队列放节点，取节点的时候怎么知道取几个同层的节点？
第一步：给队列放1个节点【3】
第二步：取出【3】，放入【3】的所有子节点【9 20】
第三步：取出【9】，放入【9】的所有子节点【】；取出【20】，放入【20】的所有子节点【15 7】
由此可知，需要循环套循环，外层循环是循环的层数，内层循环是拿到当前层的所有子节点
外层循环退出条件：队列是空
在这里插入图片描述

# 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 []
        result = []
        q = [root]
        while q:
            # 队列中有节点，代表这一层的节点数量 ,要将这一层的节点的所有子节点找到 ,先pop这一层所有节点再将所有子节点放进来,这个循环跑完，q中吧这一层原所有node清空，将这一层原本所有node的所有子节点放进队列
            # 保存这一层的节点值
            cur_level_list = []
            # 拿到这一层节点数量
            size = len(q)
            for _ in range(size):
                node = q.pop(0)
                cur_level_list.append(node.val)
                if node.left:
                    q.append(node.left)
                if node.right:
                    q.append(node.right)
            result.append(cur_level_list)
        return result