leetcode树之二叉树的层平均值

时间:2022-07-25
本文章向大家介绍leetcode树之二叉树的层平均值,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

本文主要记录一下leetcode树之二叉树的层平均值

题目

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。示例 1:输入:    3   /   9  20    /     15   7输出:[3, 14.5, 11]解释:第 0 层的平均值是 3 ,  第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。提示:    节点值的范围在32位有符号整数范围内。来源:力扣(LeetCode)链接:https://leetcode-cn.com/problems/average-of-levels-in-binary-tree著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解

/** * Definition for a binary tree node. * public class TreeNode { *     int val; *     TreeNode left; *     TreeNode right; *     TreeNode(int x) { val = x; } * } */class Solution {    public List<Double> averageOfLevels(TreeNode root) {        if (root == null) {            return Collections.emptyList();        }        List<Double> result = new ArrayList<Double>();        Queue queue = new LinkedList();        queue.offer(root);        while(!queue.isEmpty()) {            int size = queue.size();            double sum = 0;            for(int i=0; i< size; i++){                TreeNode node = (TreeNode)queue.poll();                sum += node.val;                if (node.left != null) {                    queue.offer(node.left);                }                if (node.right !=null) {                    queue.offer(node.right);                }            }            result.add(sum*1.0/size);        }        return result;    }}

小结

这里借助队列进行层次遍历,每次先记录queue的size,然后按size来poll,取出元素累加sum,然后把不为null的node.left及node.right放入到queue中,最后计算sum/size放入到result中。

doc