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

Posted 2022-08-06 HERODING23

解题思路

套用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;
    
;