题解 | #牛群平均重量# java

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 double浮点型一维数组
     */
    public double[] averageOfLevels (TreeNode root) {
        // write code here
        List<Double> averages = new ArrayList<>();
        if (root == null) return new double[0];

        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int levelSize = queue.size();
            double sum = 0;
            for (int i = 0; i < levelSize; i++) {
                TreeNode node = queue.poll();
                sum += node.val;

                if (node.left != null) queue.offer(node.left);
                if (node.right != null) queue.offer(node.right);
            }
            double average = sum / levelSize;
            averages.add(average);
        }

        double[] result = new double[averages.size()];
        for (int i = 0; i < averages.size(); i++) {
            result[i] = averages.get(i);
        }
        return result;
    }
}

该代码使用的是Java编程语言。

题目考察的是二叉树的层次遍历和求解每层节点值的平均数的问题。

在averageOfLevels方法中，首先创建一个vector<double>类型的变量averages来存储每层的平均值。

然后判断根节点是否为空，如果为空，则直接返回空的数组。

使用队列queue来进行层次遍历。将根节点入队。

然后循环，直到队列为空。在每次循环中，先获取当前层级的节点数量levelSize，并定义一个变量sum来存储该层的节点值之和。

使用一个内层循环，循环遍历当前层级的节点。将节点出队，并将节点的值加到sum上。

如果节点的左子节点不为空，则将左子节点入队；如果节点的右子节点不为空，则将右子节点入队。

内层循环结束后，计算当前层级的平均值average，将其添加到averages中。

最后，将averages转换为数组返回作为结果。