【一天一题—Day12】1161. 最大层内元素和

前往闪闪の小窝以获得更好的阅读和评论体验

一、前景提要

yesterday……hard……🤐

二、题目

1161. 最大层内元素和

难度：中等

给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层，而根节点的子节点位于第 2 层，依此类推。

请返回层内元素之和 最大 的那几层（可能只有一层）的层号，并返回其中 最小 的那个。

示例1：

输入：root = [1,7,0,7,-8,null,null]
输出：2
解释：
第 1 层各元素之和为 1，
第 2 层各元素之和为 7 + 0 = 7，
第 3 层各元素之和为 7 + -8 = -1，
所以我们返回第 2 层的层号，它的层内元素之和最大。

示例2：

输入：root = [989,null,10250,98693,-89388,null,null,null,-32127]
输出：2

提示：

• 树中的节点数在 [1, 104]范围内
• -105 <= Node.val <= 105

三、解答

1，非官方解法

BFS，一层一层遍历

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */

class Solution {
    int maxSum = Integer.MIN_VALUE;
    int minFloor = 0;

    // 一层一层遍历
    public int maxLevelSum(TreeNode root) {
        // 建一个队列用于存储每一层的所有节点
        Deque<TreeNode> queue = new ArrayDeque<>();
        queue.offer(root);
        // 用于记录层号
        int floor = 0;
        while(!queue.isEmpty()){
            floor++;
            // 用于记录节点个数
            int size = queue.size();
            // 用于记录层和
            int sum = 0;
            // 当队列里不止一个节点时
            while(size-- != 0){
                TreeNode cur = queue.poll();
                // 求和
                sum += cur.val;
                // 节点不为空时把它们加入队列等着下次一起求和
                if(cur.left != null) queue.offer(cur.left);
                if(cur.right != null) queue.offer(cur.right);
            }
            // 不用等于号，这样当出现多个相同的sum值的时候，minFloor的值就永远最小
            if(sum > maxSum){
                maxSum = sum;
                minFloor = floor;
            }
        }
        return minFloor;
    }
}

2，官方解法

官方解法
BFS和DFS