637. 二叉树的层平均值
作者:互联网
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.
示例 1:
输入: 3 / \ 9 20 / \ 15 7 输出: [3, 14.5, 11] 解释: 第0层的平均值是 3, 第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].
注意:
- 节点值的范围在32位有符号整数范围内。
class Solution { public List<Double> averageOfLevels(TreeNode root) { List<Double> res = new ArrayList<>(); Queue<TreeNode> queue = new LinkedList<>(); if(root == null) return res; queue.add(root); while(!queue.isEmpty()){ int size = queue.size(); double sum = 0.0; for(int i = 0; i < size; i++){ TreeNode node = queue.poll(); sum += node.val; if(node.left!=null) queue.add(node.left); if(node.right!=null) queue.add(node.right); } res.add(sum / size); } return res; } }
标签:node,平均值,637,res,queue,add,二叉树,null,size 来源: https://www.cnblogs.com/Roni-i/p/10510078.html