题解 | #重量级的一层#

重量级的一层

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

题目考察的知识点是:

本题主要考察遍历和二叉树知识。

题目解答方法的文字分析:

本题要求求出二叉树中和最大的那一层。可以层序遍历这个二叉树,遍历过程中维护重量级的那一层的数值和索引。树的层序遍历一般使用一个队列,先将头节点放入队列中,然后对队列中的每个元素,首先访问这个元素,然后将该元素的左右孩子结点放入队列中,直至队列中为空。

本题解析所用的编程语言:

java语言。

完整且正确的编程代码:

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
        ArrayList<Integer> arrayList = new ArrayList<>();
        ArrayList<ArrayList<Integer>> arrayLists = level(root);
        for (int i = 0; i < arrayLists.size(); i++) {
            int sum = 0;
            for (int j = 0; j < arrayLists.get(i).size(); j++) {
                sum = sum + arrayLists.get(i).get(j);
            }
            arrayList.add(sum);
        }
        int max = 0;
        int index = 0;
        for (int i = 0; i < arrayList.size(); i++) {
            if (arrayList.get(i) >= max) {
                max = arrayList.get(i);
                index = i;
            }
        }
        return index + 1;
    }

    public ArrayList<ArrayList<Integer>> level(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        ArrayList<ArrayList<Integer>> arrayLists = new ArrayList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            int size = queue.size();
            ArrayList<Integer> arr = new ArrayList<>();
            for (int i = 0; i < size; i++) {
                TreeNode q = queue.remove();
                arr.add(q.val);
                if (q.left != null) {
                    queue.add(q.left);
                }
                if (q.right != null) {
                    queue.add(q.right);
                }
            }
            arrayLists.add(arr);
        }
        return arrayLists;
    }
}

#题解#
全部评论

相关推荐

11-27 17:08
已编辑
牛客_产品运营部_私域运营
腾讯 普通offer 24k~26k * 15,年包在36w~39w左右。
点赞 评论 收藏
分享
11-03 14:38
重庆大学 Java
AAA求offer教程:我手都抬起来了又揣裤兜了
点赞 评论 收藏
分享
我在朝九晚六双休的联想等你:如果我是你,身体素质好我会去参军,然后走士兵计划考研211只需要200多分。
点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务