C++刷leetcode（1161. 最大层内元素和）【广度优先搜索】

糖逗

题目描述：

1. 给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层，而根节点的子节点位于第 2 层，依此类推。
   
2. 
   
3. 请你找出层内元素之和 最大 的那几层（可能只有一层）的层号，并返回其中 最小 的那个。
   
4. 
   
5. 
   
6. 
   
7. 示例：
   
8. 
   
9. 
   
10. 
    
11. 输入：[1,7,0,7,-8,null,null]
    
12. 输出：2
    
13. 解释：
    
14. 第 1 层各元素之和为 1，
    
15. 第 2 层各元素之和为 7 + 0 = 7，
    
16. 第 3 层各元素之和为 7 + -8 = -1，
    
17. 所以我们返回第 2 层的层号，它的层内元素之和最大。
    
18. 
    
19. 
    
20. 提示：
    
21. 
    
22. 树中的节点数介于 1 和 10^4 之间
    
23. -10^5 <= node.val <= 10^5

复制代码

1. /**
   
2. * Definition for a binary tree node.
   
3. * struct TreeNode {
   
4. *     int val;
   
5. *     TreeNode *left;
   
6. *     TreeNode *right;
   
7. *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
   
8. * };
   
9. */
   
10. class Solution {
    
11. public:
    
12.     int maxLevelSum(TreeNode* root) {
    
13.         //广度优先搜素
    
14.         queue<TreeNode*> temp;
    
15.         temp.push(root);
    
16.         vector<pair<int, int> > store;
    
17.         int i = 0;
    
18.         while(!temp.empty()){
    
19.             i++;
    
20.             int sum = 0;
    
21.             int len = temp.size();
    
22.             for(int i = 0; i < len; i++){
    
23.                 TreeNode* temp1 = temp.front();
    
24.                 temp.pop();
    
25.                 sum += temp1 -> val;
    
26.                 if(temp1 -> left) temp.push(temp1 -> left);
    
27.                 if(temp1 -> right) temp.push(temp1 -> right);
    
28.             }
    
29.             store.push_back(make_pair(sum, i));
    
30.         }
    
31.         sort(store.begin(), store.end());
    
32.         return store[store.size()-1].second;
    
33.     }
    
34. };

复制代码