leetCode_637_AverageOfLevelsInBinaryTree

最新推荐文章于 2024-09-01 09:33:26 发布

原创最新推荐文章于 2024-09-01 09:33:26 发布 · 178 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

Java 同时被 2 个专栏收录

48 篇文章

订阅专栏

算法（algorithm）

12 篇文章

订阅专栏

本文介绍了一种算法，该算法接收一棵非空二叉树作为输入，并返回一个数组，数组中的每个元素代表二叉树每一层节点值的平均数。通过广度优先搜索策略实现，确保了对二叉树每一层节点的遍历并计算平均值。

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

题目描述

Given a non-empty binary tree, return the average value of the nodes on each level in the form of an array.

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
import java.util.Queue;
class Solution {
    public List<Double> averageOfLevels(TreeNode root) {
        if (root == null)
            return null;
        List<Double> list = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        list.add((double)root.val);
        queue.add(root);
        TreeNode last = root;
        TreeNode nextLast = root;
        double averageSum  = 0;
        int count = 0;
        while (queue.size() != 0) {
            TreeNode temp = queue.poll();

            if (temp.left != null) {
                nextLast = temp.left;
                queue.add(temp.left);
                averageSum += temp.left.val;
                count ++;
            }

            if (temp.right != null) {
                nextLast = temp.right;
                queue.add(temp.right);
                averageSum += temp.right.val;
                count ++;
            }

            if (temp == last) {
                if (nextLast != last) {                    
                     list.add(averageSum / count);
                     averageSum = 0;
                     count = 0;
                     last = nextLast;
                }   
            }  
        }
        return list;

    }
}