637. Average of Levels in Binary Tree

最新推荐文章于 2024-08-26 10:59:05 发布

原创最新推荐文章于 2024-08-26 10:59:05 发布 · 167 阅读

0 ·

CC 4.0 BY-SA版权

Tree 同时被 2 个专栏收录

84 篇文章

订阅专栏

我的Leetcode专栏

77 篇文章

订阅专栏

本文介绍了一种计算二叉树每个层级节点平均值的高效算法。通过使用队列进行层次遍历，该算法能在O(n)的时间复杂度内完成，其中n为树中节点的数量。文章提供了两种实现方式，第一种使用单一队列和循环更新队列元素，第二种采用双队列交替的方式进行层次遍历。这两种方法均展示了如何在遍历过程中计算并存储各层级的平均值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

在这里插入图片描述

方法思路

Approach1:

class Solution {
    //Runtime: 4 ms, faster than 99.22% of Java
    //Memory Usage: 42.1 MB, less than 62.80% 
    public List<Double> averageOfLevels(TreeNode root) {
        List<Double> res = new LinkedList<>();
        if(root == null) return res;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(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.offer(node.left);
                if(node.right != null)
                    queue.offer(node.right);
            }
                res.add(sum / size);
            }
            return res;
        }
    }

Approach2:

public class Solution {
    public List < Double > averageOfLevels(TreeNode root) {
        List < Double > res = new ArrayList < > ();
        Queue < TreeNode > queue = new LinkedList < > ();
        queue.add(root);
        while (!queue.isEmpty()) {
            long sum = 0, count = 0;
            Queue < TreeNode > temp = new LinkedList < > ();
            while (!queue.isEmpty()) {
                TreeNode n = queue.remove();
                sum += n.val;
                count++;
                if (n.left != null)
                    temp.add(n.left);
                if (n.right != null)
                    temp.add(n.right);
            }
            queue = temp;
            res.add(sum * 1.0 / count);
        }
        return res;
    }
}