Find Leaves of Binary Tree

最新推荐文章于 2024-12-09 16:25:11 发布

weixin_33725807

最新推荐文章于 2024-12-09 16:25:11 发布

阅读量496

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/sherylwang/p/5667583.html

本文介绍了一种有趣且实用的二叉树处理算法——逐步去除叶子结点并收集其值，直至树为空。通过递归计算二叉树结点高度，确保按高度顺序输出结点值。该算法的时间复杂度为O(n)，空间复杂度为O(logn)。

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

Given a binary tree, collect a tree's nodes as if you were doing this: Collect and remove all leaves, repeat until the tree is empty.

Example:
Given binary tree

Returns [4, 5, 3], [2], [1].

Explanation:

1. Removing the leaves [4, 5, 3] would result in this tree:

          1
         / 
        2

2. Now removing the leaf [2] would result in this tree:

3. Now removing the leaf [1] would result in the empty tree:

[]

Returns [4, 5, 3], [2], [1].

很有意思的一道题，主要是一步一步去除叶子结点，然后返回结果。其实这题考的是二叉树结点的高度。高度最低的先输出，同等高度的一块输出。略微修改下递归球二叉树高度的方法就可以，同时可以根据高度，确定加结果的地方。空间复杂度O(logn),时间复杂度位O(n),每个节点都需要处理一次。代码如下：

class Solution(object):
    def findLeaves(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if not root:
            return []
        res = []
        self.dfs(root, res)
        return res
    def dfs(self, root, res):
        if not root:
            return -1
        left = self.dfs(root.left, res)
        right = self.dfs(root.right, res)
        level = max(left, right) + 1
        if len(res) == level:
            res.append([root.val])
        else:
            res[level].append(root.val)
        root.left = root.right = None
        return level