Leet Code102 Binary Tree Level Order Traversal

最新推荐文章于 2019-06-11 09:41:05 发布

原创最新推荐文章于 2019-06-11 09:41:05 发布 · 131 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Python #leetcode

leetcode 专栏收录该内容

32 篇文章

订阅专栏

本文介绍了二叉树的层次遍历算法，包括使用双端队列实现的层次遍历和利用深度优先遍历记录深度的方法。通过实例解析，详细阐述了如何逐层访问二叉树的所有节点。

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

给定一个二叉树，返回其按层次遍历的节点值。（即逐层地，从左到右访问所有节点）。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

思路一：层次遍历
从根节点开始，一层一层的处理节点，
使用双端队列存放每层的节点，visit过一个节点就弹出该节点。
使用了for loop遍历一层的节点，后续的插入下一层节点不会影响到当前层的遍历。

class Solution:
    def levelOrder(self, root):
        if not root: return []
        res = []
        queue = [root]
        while queue:
            current_level = []
            for _ in range(len(queue)):
                node = queue.pop(0)
                current_level.append(node.val)
                if node.left: queue.append(node.left)
                if node.right: queue.append(node.right)
            res += current_level,
        return res

思路二：深度优先遍历，记录深度,

class Solution:
    def levelOrder(self, root):
        if not root: return []
        self.res = []
        dfs(root, 0)
        return self.res

    def dfs(self, node, level):
        if not node: return
        if len(self.res) < level+1:
            self.res += [],
        self.res[level].append(node.val)

        self.dfs(node.left, level+1)
        self.dfs(node.right, level+1)