leetcode-层数最深叶子节点的和

87 阅读1分钟

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

题目描述

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

示例 1:

1302示例1.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

思路

本题要求出层数最深的叶子节点的和,其实就是求最下一层节点的和,因为既然是最下面一层的节点,就一定是叶子节点了。我们可以安全层序遍历的方式,求出每1层的和,那么最后1次求出的和就是层数最深的叶子节点的和。

层序遍历的方法:

  1. 定义1个队列
  2. 将根节点加入队列
  3. 如果队列非空,进入while循环
  4. 获取队列的大小,依次取出
  5. 取出节点如果存在左子节点或者右子节点,将子节点加入队列中

不难发现,通过这样遍历,每次遍历的节点都是在同一层的,第5步讲子节点加入队列是给下一次循环遍历使用的,本次循环遍历通过第4步控制了大小,不会遍历到下一层的节点。

本题就是在循环遍历的时候,对每一层进行求和,这样非常容易理解,但是缺陷在于对于求和了很多次,特别是深度较深的时候,前面的几次求和都是有浪费的。如果要避免这种情况,需要先遍历去到最深的层数,但是这样就要遍历2次,只是一个取舍。

Java版本代码

class Solution {
    public int deepestLeavesSum(TreeNode root) {
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        int sum = 0;
        while (!queue.isEmpty()) {
            sum = 0;
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                sum += node.val;
                if (node.left != null) {
                    queue.add(node.left);
                }
                if (node.right != null) {
                    queue.add(node.right);
                }
            }
        }
        return sum;
    }
}