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

共 2083字,需浏览 5分钟

 ·

2021-09-02 17:02

算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从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:敲击计数器

浏览 27
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报