[路飞]_LeetCode_1302. 层数最深叶子节点的和

「这是我参与2022首次更文挑战的第35天，活动详情查看：2022首次更文挑战」

题目

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

示例 1：

输入：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

来源：力扣（LeetCode）leetcode-cn.com/problems/de…

解题思路

根据题目意思，我们需要找到树的深度，然后统计最深一层所有节点值的总和。

通过深度优先遍历方式计算每层节点的和

• 将根节点的值放在第 k 层
• 将左子节点的值放在第 k + 1 层
• 将右子节点的值放在第 k + 1 层
• 统计时定义临时对象 { max_k: 0, ans: 0 } 存储最大层数 k 和第 k 层的和，如果根节点层数大于 k 则更新 k 值为当前节点的层数，如果根节点层数等于 k ，则将根节点的值累加到第 k 层的和中。

代码实现

var deepestLeavesSum = function (root) {
    //定义临时对象，统计函数需要修改该对象的值
    const obj = {
        max_k: 0,  //最大层：默认为0
        ans: 0  //节点和：默认为0
    }
    
    //统计最深叶子节点和放在obj.ans中
    getResult(root, 1, obj)
    return obj.ans
};

var getResult = function (root, k, ref) {
    //根节点为空时不处理
    if (!root) return

    //如果当前层数等于最大层数，则累加节点的值
    if (k === ref.max_k) {
        ref.ans += root.val
    }

    //如果当前层数大于最大层数，则更新最大层数，并且更新最大层数的节点和为当前节点的值
    if (k > ref.max_k) {
        ref.max_k = k
        ref.ans = root.val
    }

    //将左节点的值累加到第 k + 1 层
    getResult(root.left, k + 1, ref)
    //将右节点的值累加到第 k + 1 层
    getResult(root.right, k + 1, ref)
}

如有错误欢迎指出，欢迎一起讨论！