Advertisement
nikunjsoni

1161

Apr 24th, 2021
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.02 KB | None | 0 0
  1. /**
  2.  * Definition for a binary tree node.
  3.  * struct TreeNode {
  4.  *     int val;
  5.  *     TreeNode *left;
  6.  *     TreeNode *right;
  7.  *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
  8.  *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
  9.  *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
  10.  * };
  11.  */
  12. class Solution {
  13. public:
  14.     int maxLevelSum(TreeNode* root) {
  15.         queue<TreeNode*> q;
  16.         q.push(root);
  17.        
  18.         int level=0, ans=1, maxSum=INT_MIN;
  19.         while(!q.empty() && ++level){
  20.             int sz = q.size();
  21.             int sum = 0;
  22.             for(int i=0; i<sz; i++){
  23.                 TreeNode *node = q.front();
  24.                 q.pop();
  25.                 sum += node->val;
  26.                 if(node->left) q.push(node->left);
  27.                 if(node->right) q.push(node->right);
  28.             }
  29.             if(sum>maxSum){
  30.                 maxSum = sum;
  31.                 ans = level;
  32.             }
  33.         }
  34.         return ans;
  35.     }
  36. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement