LC 637. 二叉树的层平均值

[HealUP]

637. 二叉树的层平均值

思路：

• 层次遍历的时候，每一层求一个总和
• 该层结束，就求这一层的平均值即可

class Solution {
    public List<Double> averageOfLevels(TreeNode root) {
        Queue<TreeNode> que = new LinkedList<>();
        //判断空
        if (root != null) {
            que.offer(root);
        }
        List<Double> list = new ArrayList<>();
        while (!que.isEmpty()) {
            int len = que.size();
            double sum = 0.0;
            for (int i = 0; i < len; i++) {//这里不能用while判断 会改变len的值
                TreeNode node = que.poll();//弹出并临时记录
                sum += node.val;//求和
                //将该层的平均值存放到一维数组中
                if (node.left != null) {
                    que.add(node.left);
                }
                if (node.right != null) {
                    que.add(node.right);
                }
            }
            list.add(sum / len);
        }
        return list;
    }
}

注意：这里不能用while

不能用 while(len > 0) {
    ...
    len--；
}
因为len改变了，就无法计算sum/len了，这点要注意