1161. 最大层内元素和

最新推荐文章于 2024-04-20 18:58:37 发布

原创最新推荐文章于 2024-04-20 18:58:37 发布 · 335 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

4 篇文章

订阅专栏

本文探讨了一道经典的二叉树与BFS综合题，旨在寻找二叉树中层内元素之和最大的层号。通过详细的算法解析和C++代码示例，展示了如何使用队列进行广度优先搜索，计算每层节点值的总和，最终确定具有最大层内元素之和的层号。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层，而根节点的子节点位于第 2 层，依此类推。
请你找出层内元素之和最大的那几层（可能只有一层）的层号，并返回其中最小的那个。

示例

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

分析

这是一个典型的树与BFS综合题。通过计算树的每一层节点之和，方可回答此题。
此题涉及到了队列的BFS。
队列的作用是不仅要用于计算出当前层的节点值的和，还要保存本层节点孩子的地址。
通过while()循环，算出一层的值。

参考代码

class Solution {
public:
    int maxLevelSum(TreeNode* root) {

        if(root == nullptr) {return 0;}
        //
        int maxlevel = 1 , curlevel = 1 , summax = 1 ;
        queue<TreeNode*> q ;
        q.push(root);
        while(!q.empty())
        {
            int s = q.size();
            int cursum = 0;
            while(s)
            {
                TreeNode* font = q.front();
                q.pop();
                if(font->left != nullptr){q.push(font->left);}
                if(font->right != nullptr){q.push(font->right);}
                cursum += font->val;
                --s;
            }
            if(cursum > summax){maxlevel = curlevel ; summax = cursum ; }
            ++curlevel;
        }
        return maxlevel;
    }
};

欢迎访问我的博客，了解最新动态https://www.mckinleylu.com/