题目：637.二叉树的层平均值

最新推荐文章于 2024-09-14 18:52:56 发布

十年一觉尘与土

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量36

点赞数

分类专栏： # Java 文章标签： leetcode java

本文链接：https://blog.csdn.net/2301_76145947/article/details/129688053

版权

Java 专栏收录该内容

586 篇文章 1 订阅

订阅专栏

该文章介绍了如何使用广度优先搜索（BFS）策略解决LeetCode上的637题，即求解二叉树每一层的平均值。在Java代码中，通过创建队列存储节点并逐层遍历，计算每层节点值的总和与数量，从而得到平均值。初次提交时需注意数值溢出的问题。

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

题目来源：

leetcode题目，网址：637. 二叉树的层平均值 - 力扣（LeetCode）

解题思路：

广度优先遍历二叉树，计算层平均值即可。

解题代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<Double> averageOfLevels(TreeNode root) {//节点数目不为0
        List<Double> res=new ArrayList<>();
        Queue<TreeNode> queue=new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            double sum=0;
            int size=queue.size();
            for(int i=0;i<size;i++){
                TreeNode temp=queue.poll();
                sum+=temp.val;
                if(temp.left!=null){
                    queue.offer(temp.left);
                }
                if(temp.right!=null){
                    queue.offer(temp.right);
                }
            }
            res.add(sum/size);
        }
        return res;
    }
}

总结：

第一次提交时没有考虑到和可能超过 int 表示范围，出现错误结果。