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

题目路径

题目描述

给你一棵二叉树的根节点 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

解题思路

• max_k记录当前已经找到的当前节点的最大深度
• ans当前记录到的最大深度节点和值

代码

// 用一个技巧，一次记录两个值:
// 1个是当前已经找到的当前节点的最大深度
// 1个是当前记录到的最大深度的节点和值
var deepestLeavesSum = function(root) {
    let max_k = [0], ans = [0];
    getResult(root, 0, max_k, ans);
    return ans[0];
};

var getResult = function(root, k, max_k, ans){
    if(root == null) return ;
    if(k == max_k[0]){
        ans[0] += root.val;
    }
    if(k > max_k[0]) {
        max_k[0] = k;
        ans[0] = root.val;
    }
    getResult(root.left, k + 1, max_k, ans);
    getResult(root.right, k + 1, max_k, ans);
    return ans;
}