leetcode_1302 层数最深叶子节点的和

99 阅读1分钟

要求

给你一棵二叉树的根节点 root ,请你返回 层数最深的叶子节点的和 。

示例 1:

image.png

输入:root = [1,2,3,4,5,null,6,7,null,null,null,null,8]
输出:15

示例 2:

输入:root = [6,7,8,2,7,1,3,9,null,1,4,null,null,null,5]
输出:19

提示:

  • 树中节点数目在范围 [1, 104] 之间。
  • 1 <= Node.val <= 100

核心代码

# 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 deepestLeavesSum(self, root: Optional[TreeNode]) -> int:
        from collections import deque
        if not root:
            return 0
        queue = deque([root])
        while queue:
            res = []
            for _ in range(len(queue)):
                node = queue.popleft()
                res.append(node.val)
                if node.left:
                    queue += [node.left]
                if node.right:
                    queue += [node.right]
        return sum(res)

image.png

解题思路:这个题就是广度优先遍历,我们存储每一层的值,最终得到最后一层,将所有值加和即可。