Jenifa

Most Frequent Subtree Sum

Jun 10th, 2020
1,065
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.44 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.     private:
  14.     map<int, int> m;
  15.     vector<int> v;
  16.    
  17.        
  18.         int subtree(TreeNode* root, int& freq)
  19.         {
  20.             int left, right, sum = 0;
  21.            
  22.            if(!root)
  23.            {
  24.                return 0;
  25.            }else
  26.            {
  27.                left = subtree(root->left, freq);
  28.                right = subtree(root->right, freq);
  29.                sum = left + right + root->val;
  30.            
  31.                m[sum]++;
  32.            
  33.                if (m[sum] > freq)
  34.                  {
  35.                     freq = m[sum];
  36.            
  37.                      v.clear();
  38.            
  39.                      v.push_back(sum);
  40.                    
  41.                   } else if (m[sum] == freq)
  42.                   {
  43.                       v.push_back(sum);
  44.                    }
  45.            
  46.                }
  47.            
  48.        
  49.         return sum;
  50.     }
  51. public:
  52.     vector<int> findFrequentTreeSum(TreeNode* root)
  53.     {
  54.         int freq=0;
  55.        
  56.         subtree(root, freq);
  57.        
  58.         return v;
  59.        
  60.     }
  61. };
Advertisement
Add Comment
Please, Sign In to add comment