637.二叉树的层平均值

题目:637.二叉树的层平均值

给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10^-5 以内的答案可以被接受。

• 示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：[3.00000,14.50000,11.00000]
解释：第 0 层的平均值为 3,第 1 层的平均值为 14.5,第 2 层的平均值为 11 。
因此返回 [3, 14.5, 11] 。

• 示例 2：

输入：root = [3,9,20,15,7]
输出：[3.00000,14.50000,11.00000]

• 提示：

树中节点数量在 [1, 10^4] 范围内
-2^31 <= Node.val <= 2^31 - 1

思路

• 利用广度优先搜索 (BFS) 遍历二叉树的每一层。

• 对每一层，累加节点的值，然后计算平均值并存储在结果列表中。

• 时间复杂度:O(n)

• 空间复杂度:O(n)

代码

public List<Double> averageOfLevels(TreeNode root) {
    List<Double> result = new LinkedList<>();
    if (root == null) {
        return result;
    }
    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);
            }
        }
        // 累加节点的值，然后计算平均值并存储在结果列表中
        result.add(sum / levelSize);
    }
    return result;
}