leetcode 637. Average of Levels in Binary Tree（二叉树每层的平均数）

蓝羽飞鸟

已于 2022-09-02 15:35:43 修改

阅读量107

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签：算法 leetcode

于 2020-12-11 21:37:56 首次发布

本文链接：https://blog.youkuaiyun.com/level_code/article/details/111053238

leetcode 专栏收录该内容

490 篇文章

订阅专栏

本文介绍了一种算法，用于计算二叉树各层节点值的平均数，并以数组形式返回结果。通过广度优先搜索（BFS）遍历树的每一层，计算节点值总和后除以该层节点数量得到平均值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a non-empty binary tree, return the average value of the nodes on each level in the form of an array.
Example 1:

Input:
    3
   / \
  9  20
    /  \
   15   7

Output: [3, 14.5, 11]
Explanation:
The average value of nodes on level 0 is 3, on level 1 is 14.5, and on level 2 is 11. Hence return [3, 14.5, 11].
Note:
The range of node’s value is in the range of 32-bit signed integer.

计算二叉树每层的平均值，返回list

思路：
BFS，对每层节点值求和再求平均
注意和要用long类型，用int类型在相加的时候可能会溢出，保存时cast到double。或者直接用double型

public List<Double> averageOfLevels(TreeNode root) {
    Queue<TreeNode> queue = new LinkedList<>();
    List<Double> res = new ArrayList<>();
    queue.offer(root);
    
    while(!queue.isEmpty()) {
        double val = 0.0;
        int size = queue.size();
        for(int i = 0; i < size; i++) {
            TreeNode cur = queue.poll();
            val += cur.val;
            if(cur.left != null) queue.offer(cur.left);
            if(cur.right != null) queue.offer(cur.right);
        }
        res.add(val/size);
    }
    return res;
}