LeetCode 637. 二叉树的层平均值(Average of Levels in Binary Tree)

Posted hglibin

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode 637. 二叉树的层平均值(Average of Levels in Binary Tree)相关的知识,希望对你有一定的参考价值。

637. 二叉树的层平均值
637. Average of Levels in Binary Tree

LeetCode637. Average of Levels in Binary Tree

题目描述
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.

示例 1:
输入:

    3
   /   9  20
    /     15   7

输出: [3, 14.5, 11]
解释:
第 0 层的平均值是 3,第 1 层是 14.5,第 2 层是 11.因此返回 [3, 14.5, 11].

注意:

  1. 节点值的范围在32位有符号整数范围内。

Java 实现

import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}

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

相似题目

参考资料

以上是关于LeetCode 637. 二叉树的层平均值(Average of Levels in Binary Tree)的主要内容,如果未能解决你的问题,请参考以下文章

637. 二叉树的层平均值(Python)

每日一扣637. 二叉树的层平均值

637. 二叉树的层平均值

二叉树层序遍历集合

637. Average of Levels in Binary Tree(一棵树每层节点的平均数)(二叉树的层序遍历)

leetcode记录