365天挑战LeetCode1000题——Day 045 最大层内元素和层次遍历

原创已于 2022-08-01 08:04:09 修改 · 90 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

于 2022-07-31 09:57:42 首次发布

LeetCode千题之路专栏收录该内容

124 篇文章

订阅专栏

本文介绍了一种使用广度优先搜索算法解决二叉树中最大层内元素和问题的C++实现。通过维护当前层节点和总和，找到最大层的和及其对应的层级。适用于面试题或理解递归遍历的应用。

1161. 最大层内元素和

在这里插入图片描述

代码实现（自解）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int maxLevelSum(TreeNode* root) {
        int maxSum = INT_MIN, maxLevel = 0;
        queue<TreeNode*> _queue;
        int curSum = 0, curLevel = 0, sz = 0;
        TreeNode* curNode = NULL;
        _queue.push(root);
        while (!_queue.empty()) {
            sz = _queue.size();
            curSum = 0;
            curLevel++;
            while (sz--) {
                curNode = _queue.front();
                _queue.pop();
                curSum += curNode->val;
                if (curNode->left) _queue.push(curNode->left);
                if (curNode->right) _queue.push(curNode->right);
            }
            if (curSum > maxSum) {
                maxSum = curSum;
                maxLevel = curLevel;
            }
        }
        return maxLevel;
    }
};