给你一个二叉树的根节点
root
。设根节点位于二叉树的第1
层,而根节点的子节点位于第2
层,依此类推。请返回层内元素之和 最大 的那几层(可能只有一层)的层号,并返回其中 最小 的那个。
示例 1:
输入:root = [1,7,0,7,-8,null,null] 输出:2 解释: 第 1 层各元素之和为 1, 第 2 层各元素之和为 7 + 0 = 7, 第 3 层各元素之和为 7 + -8 = -1, 所以我们返回第 2 层的层号,它的层内元素之和最大。示例 2:
输入:root = [989,null,10250,98693,-89388,null,null,null,-32127] 输出:2
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {number}
*/
var maxLevelSum = function(root) {
let arr = [];
function dfs(node,depth){
if(!node) return;
arr[depth] = arr[depth] || [];
arr[depth].push(node.val);
dfs(node.left,depth + 1);
dfs(node.right,depth + 1);
}
dfs(root,0);
let ans = [];
for(let i = 0;i < arr.length;i++){
sum = arr[i].reduce((sum,cur) => sum + cur);
ans.push(sum);
}
let max = Math.max(...ans);
return ans.indexOf(max) + 1;
};
标签:node,层内,arr,right,leetcode1161,元素,depth,null,root
From: https://blog.csdn.net/Turboyiyi/article/details/143854917