题解 | #重量级的一层# Java

代码中的 maxLevelSum 方法接受一个树节点 root 作为输入，并返回最大总重量的层数。

首先，我们对特殊情况进行处理，如果 root 为空树，则返回 0。

然后，我们初始化最大总重量的层数 maxLevel 为 1，最大总重量 maxSum 为根节点的值 root.val，当前层数 level 为 1。

我们使用一个队列 queue 实现层序遍历，并将根节点入队。

接下来，我们进行层序遍历，每次遍历一层的节点。我们在遍历过程中，使用 levelSum 记录当前层的总重量，并根据节点的左右子节点将它们加入队列中。

在每轮遍历结束后，我们更新最大总重量层及总重量的值。如果当前层的总重量 levelSum 大于 maxSum，则更新 maxSum 为 levelSum 并 maxLevel 为 level。

最后，返回最大总重量的层数 maxLevel 即可。

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类
     * @return int整型
     */
    public int maxLevelSum (TreeNode root) {
        if (root == null) {
            return 0;
        }

        int maxLevel = 1; // 最大总重量的层数
        int maxSum = root.val; // 最大总重量
        int level = 1; // 当前层数

        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root); // 将根节点入队

        while (!queue.isEmpty()) {
            int levelSum = 0; // 当前层的总重量
            int levelSize = queue.size();

            // 遍历当前层的所有节点
            for (int i = 0; i < levelSize; i++) {
                TreeNode node = queue.poll();
                levelSum += node.val;

                // 将当前节点的左右子节点入队
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }

            // 更新最大总重量层及总重量
            if (levelSum > maxSum) {
                maxSum = levelSum;
                maxLevel = level;
            }

            level++; // 进入下一层
        }

        return maxLevel;
    }
}