题解 | #牛群平均重量#

牛群平均重量

https://www.nowcoder.com/practice/9b826d0a84034e8e8afd4c451f7d34e0

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return double浮点型一维数组
     */
    public double[] averageOfLevels (TreeNode root) {
        // write code here
        List<List<Integer>> lists = new ArrayList<>();
        levelOrder(root, 0, lists);
        double[] ans = new double[lists.size()];
        int i = 0;
        for (List<Integer> list : lists) {
            ans[i++] = (list.stream().mapToInt(v -> v).sum() * 1.0) / list.size();
        }
        return ans;
    }
    private void levelOrder(TreeNode root, int level, List<List<Integer>> lists) {
        if (root == null) return;
        if (level == lists.size()) lists.add(new ArrayList<>());
        lists.get(level).add(root.val);
        levelOrder(root.left, level + 1, lists);
        levelOrder(root.right, level + 1, lists);
    }
}
  • 根据题意,可以基于二叉树的层序遍历求解。
  • 定义一个集合 List<List<Integer>> lists,收集二叉树每一层节点的值,然后遍历 lists 求每一层的平均值
#二叉树的层序遍历#
线性表基础 文章被收录于专栏

链表、递归、栈

全部评论

相关推荐

05-29 22:11
门头沟学院 Java
Elastic90:抛开学历造假不谈,这公司的招聘需求也挺怪的,Java开发还要求你有图文识别、移动端开发和c++的经验,有点逆天了。
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-02 17:28
25届每天都在焦虑找工作的事情0offer情绪一直很低落硬撑着面了一个岗位岗位有应酬的成分面试的时候hr给我出各种场景题问的问题比较犀利&nbsp;有点压力面的感觉感觉有点回答不上来本来就压抑的情绪瞬间爆发了呢一瞬间特别想哭觉得自己特别没用没绷住掉眼泪了事后想想觉得自己挺有病的&nbsp;真的破大防了
喜欢唱跳rap小刺猬...:我觉得没关系吧,之前有一次面试leader给我压力面,我顶住了压力,结果入职的时候发现组里氛围很差,果断跑路。其实从面试就能大概看出组的情况,面试体验好的组倒是不一定好,但是面试体验不好的组。。。就很难说
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务