L75-39（Leetcode1161最大层内元素和）

从月亮走向月亮7

于 2024-03-14 09:54:45 发布

阅读量83

点赞数 2

文章标签：算法 leetcode 职场和发展

本文链接：https://blog.csdn.net/stacey777/article/details/136699531

版权

该篇文章介绍了如何使用Java实现一个名为Solution的类，其中包含一个方法maxLevelSum，用于计算给定二叉树中各层的最大和，采用广度优先搜索(BFS)策略。

摘要由CSDN通过智能技术生成

代码：

/**
 * 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 {
    public int maxLevelSum(TreeNode root) {
        if(root==null)return 0;
        Queue<TreeNode> q = new LinkedList<>();
        q.offer(root);
        int max = root.val;
        int layer = 0;
        int res = 1;
        while(!q.isEmpty()){
            layer++;
            int size = q.size();
            int curSum = 0;
            for(int i=0;i<size;i++){
                TreeNode curNode = q.poll();
                curSum += curNode.val;
                if(curNode.left!=null)q.offer(curNode.left);
                if(curNode.right!=null)q.offer(curNode.right);
            }
            if(curSum>max){
                max = curSum;
                res = layer;
            }

        }
        return res;
    }
}