题目:
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.
示例 1:
输入: 3 / \ 9 20 / \ 15 7 输出: [3, 14.5, 11] 解释: 第0层的平均值是 3, 第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].
注意:
- 节点值的范围在32位有符号整数范围内。
解题思路:
使用bfs实现每层的平均值计算。
代码实现:
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public List<Double> averageOfLevels(TreeNode root) { Queue<TreeNode> queue = new LinkedList<TreeNode>(); List<Double> res = new LinkedList<Double>(); if (root == null) return res; queue.offer(root); while (!queue.isEmpty()) { int num = queue.size(); double sum = 0; for (int i = num; i > 0; i --) { TreeNode p = queue.poll(); sum += p.val; if (p.left != null) queue.offer(p.left); if (p.right != null) queue.offer(p.right); } res.add(sum / (double) num); } return res; } }