题解 | #重量级的一层#

重量级的一层

https://www.nowcoder.com/practice/193372871b09426ab9ea805f0fd44d5c

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 int整型
     */
    public int maxLevelSum (TreeNode root) {
        // write code here
        List<Long> list = new ArrayList<>();
        levelOrder(root, 0, list);
        int size = list.size(), ans = size;
        long max = list.get(size - 1);
        for (int i = size - 2; i >= 0; i--) {
            long x = list.get(i);
            if (x > max) {
                max = x;
                ans = i + 1;
            }
        }
        return ans;
    }
    private void levelOrder(TreeNode root, int level, List<Long> list) {
        if (root == null) return;
        if (level == list.size()) list.add(0L);
        list.set(level, list.get(level) + root.val);
        levelOrder(root.left, level + 1, list);
        levelOrder(root.right, level + 1, list);
    }
}
  • 利用二叉树的层序遍历,计算二叉树每一层节点数值的累加和,将每一层的累加和存储到一个List集合中
  • 逆序遍历List集合,确定最大值以及所在层
#二叉树的层序遍历#
线性表基础 文章被收录于专栏

链表、递归、栈

全部评论

相关推荐

06-18 15:03
门头沟学院 Java
至少实习看起来比去年好?问了下群里的同学和身边的同学,人均有offer。有的还有好几个大厂offer
菜鸟1973:上一年暑期也是人均大厂实习offer,结果秋招跟不招人一样,大部分都转正了
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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