​LeetCode刷题实战366：寻找二叉树的叶子节点

算法的重要性，我就不多说了吧，想去大厂，就必须要经过基础知识和业务逻辑面试+算法面试。所以，为了提高大家的算法能力，这个公众号后续每天带大家做一道算法题，题目就从LeetCode上面选 ！

今天和大家聊的问题叫做 寻找二叉树的叶子节点，我们先来看题面：

https://leetcode-cn.com/problems/find-leaves-of-binary-tree/

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.

给你一棵二叉树，请按以下要求的顺序收集它的全部节点：

1. 依次从左到右，每次收集并删除所有的叶子节点

2. 重复如上过程直到整棵树为空

示例

解题

https://zhuanlan.zhihu.com/p/96255905

我们发现根据子树的高度来分组的

比如示例中 4, 5, 3高度都是0，2是1，1是2

class Solution:
    def findLeaves(self, root: TreeNode) -> List[List[int]]:
        res = []

        def helper(root):
            if not root:
                return 0
            left = helper(root.left)
            right = helper(root.right)
            height = max(left, right)
            if len(res) == height:
                res.append([])
            res[height].append(root.val)
            return height + 1
        helper(root)

        return res

好了，今天的文章就到这里，如果觉得有所收获，请顺手点个在看或者转发吧，你们的支持是我最大的动力 。

上期推文：

LeetCode1-360题汇总，希望对你有点帮助！

LeetCode刷题实战361：轰炸敌人

LeetCode刷题实战362：敲击计数器