637. 二叉树的层平均值

637. 二叉树的层平均值：

题目链接 ：637. 二叉树的层平均值

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

---

思路：

1、带层值记录的BFS

• 利用队列存储根节点，找到下一层其子节点加入队列。
• 根据队列中存储节点数确定每层子节点个数。
• 将每层节点按照广度优先顺序加入到子集合。
• 每层中计算节点值的平均值加入到结果集合中

2. 时间复杂度：O(n+h)=O(n) n为层序遍历遍历节点数量，h为二叉树高度，进行O(h)次计算。
3. 空间复杂度：O(n) n为集合列表中元素个数，也为n。

---

AC代码：

1. BFS

 class Solution {
    public List<Double> averageOfLevels(TreeNode root) {
        List<Double> res = new LinkedList<>();
        if (root == null) return res;

        Queue<TreeNode> q = new LinkedList<>();
        q.offer(root);
        while (!q.isEmpty()) {
            int size = q.size();
            
            double sum = 0;
            for (int i = 0; i < size; i++) {
                TreeNode cur = q.poll();
                if (cur.left != null) {
                    q.offer(cur.left);
                }
                if (cur.right != null) {
                    q.offer(cur.right);
                }
                sum += cur.val;
            }
            
            res.add(1.0*sum/size);
        }

        return res;
    }
}