题目链接

最大层内元素和

题目描述

注意点

• 返回层内元素之和 最大 的那几层（可能只有一层）的层号，并返回其中 最小 的那个
• 树中的节点数在 [1, 10000]范围内
• -10^5 <= Node.val <= 10^5

解答思路

• 广度优先遍历树，使用队列存储每一层的节点，遍历一层时，计算该层元素之和的同时将下一层的节点写入到队列中，如果某一层元素和大于此时的最大层内元素和，则将结果更新为该层，同时将最大层内元素和更新为该层元素和

代码

/*** Definition for a binary tree node.* public class TreeNode {*     int val;*     TreeNode left;*     TreeNode right;*     TreeNode() {}*     TreeNode(int val) { this.val = val; }*     TreeNode(int val, TreeNode left, TreeNode right) {*         this.val = val;*         this.left = left;*         this.right = right;*     }* }*/
class Solution {public int maxLevelSum(TreeNode root) {int res = 0;int maxVal = Integer.MIN_VALUE;int depth = 0;Deque<TreeNode> deque = new ArrayDeque<>();deque.offerLast(root);while (!deque.isEmpty()) {depth++;int size = deque.size();int sum = 0;for (int i = 0; i < size; i++) {TreeNode node = deque.pollFirst();sum += node.val;if (node.left != null) {deque.offerLast(node.left);}if (node.right != null) {deque.offerLast(node.right);}}if (sum > maxVal) {res = depth;maxVal = sum;}}return res;}
}

关键点

• 广度优先遍历的思想