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

最新推荐文章于 2022-01-23 18:17:58 发布

捕若审若判若

最新推荐文章于 2022-01-23 18:17:58 发布

阅读量119

点赞数

分类专栏：刷题文章标签：二叉树 java leetcode

本文链接：https://blog.csdn.net/qq_42411921/article/details/111540424

版权

刷题专栏收录该内容

44 篇文章 0 订阅

订阅专栏

本文介绍Leetcode637题，使用BFS方法解决二叉树各层平均值问题。通过队列进行层次遍历，计算每层节点的和并求平均，将结果存入集合中，整体时间复杂度为O(n)。

摘要由CSDN通过智能技术生成

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

##Tree##

本题求二叉树每层的结点平均值

可以采用DFS、BFS

这里采用BFS层次遍历

队列q中元素为某一层的结点，count = q.size()代表队列中元素的个数
循环count次，遍历队列q中所有结点，求出该层所有结点的和sum，并且对每个结点都需要将其左右儿子加入队列中
将每层的平均值sum/count加入到结果集合中

时间复杂度： O(n)

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