LeetCode 1161 最大层内元素和[BFS 二叉树] HERODING的LeetCode之路

解题思路

套用BFS的模板可以轻松解决该题，首先定义队列，用于按层顺序存储节点，然后遍历在队列中的每一层，统计总和，并更新最大总和所在层数，如此直到队列为空，也说明二叉树已经遍历完毕，代码如下：

代码

/**
 * 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) {
        queue<TreeNode*> q;
        int maxNum = INT_MIN;
        int layer = 1;
        int cur = 0;
        q.emplace(root);
        while(!q.empty()) {
            cur ++;
            int n = q.size();
            int total = 0;
            for(int i = 0; i < n; i ++) {
                TreeNode* temp = q.front();
                total += temp->val;
                if(temp->left != nullptr) {
                    q.emplace(temp->left);
                }
                if(temp->right != nullptr) {
                    q.emplace(temp->right);
                }
                q.pop();
            }
            if(total > maxNum) {
                maxNum = total;
                layer = cur;
            }
        }
        return layer;
    }
};