最大层内元素和

news2024/12/24 21:10:59

题目链接

最大层内元素和

题目描述

注意点

返回层内元素之和最大的那几层（可能只有一层）的层号，并返回其中最小的那个
树中的节点数在 [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;
    }
}