携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第22天，点击查看活动详情

刷题的日常-2022年8月17日

一天一题，保持脑子清爽

层数最深叶子节点的和

来自leetcode的 1302 题，题意如下：

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

示例如下：

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

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

理解题意

我们可以从题意中提取的条件如下：

• 题目给定一棵树
• 要求我们汇总最后一层的节点的总和
  • 这里不能简单的汇总叶子节点，因为叶子节点不一定是最后一层
  • 要找层数最大的一层做相加
• 返回最后一层的总和

做题思路

这里可以有两种思路：深度遍历和广度遍历

深度遍历

深度遍历其实就是一开始就往下走，遇到层数更大的就更新汇总的值为当前节点的值。如果层数相同，就加上原先的已有的值，最后返回即可。

广度遍历

广度遍历相当于层序遍历，我们可以用队列保存每一层的节点，并且在队列不为空的时候一直往里加数据，直到队列为空，则层序遍历完成：

• 开辟一个队列用来保存每一层的节点
• 遍历的过程中，如果下一层有节点，则将节点放进队列中
• 一层遍历完成之后，下一层也放进队列里面了
• 我们拿到当前层的节点数量，将节点一个个弹出队列
• 在弹出的过程中汇总当前层的结果，直接覆盖原结果就行
• 最终的结果就是最后一层的值
• 直接返回即可

代码实现

代码实现如下：

public class Solution {
    public int deepestLeavesSum(TreeNode root) {
        int result = root.val;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int size;

        while ((size = queue.size()) > 0) {
            result = 0;
            while (size-- > 0) {
                TreeNode node = queue.poll();
                result += node.val;
                if (node.left != null) {
                    queue.add(node.left);
                }
                if (node.right != null) {
                    queue.add(node.right);
                }
            }
        }
        return result;
    }
}