题解 | #重量级的一层#

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;
    }
}

本题知识点分析：

1.二叉树层序遍历

2.集合存取

3.数学模拟

本题解题思路分析：

1.利用二叉树层序遍历

2.记录每一层的和，并与max进行比较，然后记录当前最大层的索引也就是result

3.最后返回最大层的索引result即可

本题使用编程语言：Java