最大层内元素和

/*
 * 思路：
 * 虽然Java Streams不是特别适合处理树结构，但我们依然可以
 * 使用广度优先搜索（BFS）来收集每一层的节点，然后使用
 * Stream API来计算每层的和。
 */

import java.util.*;
import java.util.stream.*;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

class Solution {
    public int maxLevelSum(TreeNode root) {
        if (root == null) return 0;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        int maxSum = Integer.MIN_VALUE;
        int maxLevel = 1;
        int currentLevel = 1;
        while (!queue.isEmpty()) {
            int levelSize = queue.size();
            int levelSum = queue.stream().limit(levelSize).mapToInt(node -> node.val).sum();
            if (levelSum > maxSum) {
                maxSum = levelSum;
                maxLevel = currentLevel;
            }
            for (int i = 0; i < levelSize; i++) {
                TreeNode node = queue.poll();
                if (node.left != null) queue.offer(node.left);
                if (node.right != null) queue.offer(node.right);
            }
            currentLevel++;
        }
        return maxLevel;
    }
}