Leetcode508. 出現次數最多的子樹元素和

連結：https://leetcode-cn.com/problems/most-frequent-subtree-sum
給你一個二叉樹的根結點，請你找出出現次數最多的子樹元素和。一個結點的「子樹元素和」定義為以該結點為根的二叉樹上所有結點的元素之和（包括結點本身）。

你需要返回出現次數最多的子樹元素和。如果有多個元素出現的次數相同，返回所有出現次數最多的子樹元素和（不限順序）。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
//按照第二個值進行排序
   static bool cmp(pair<int,int> &a,pair<int,int> &b){
        return a.second > b.second;
    }
//後序遍歷模板，並且記錄值出現的次數
    int dfs(TreeNode *root,unordered_map<int,int> &m){
        int sum;
        if(!root) return 0;
        int l = dfs(root->left,m);
        int r = dfs(root->right,m);
        sum=root->val+r+l;
        m[sum]++;
        return sum;
    }
    vector<int> findFrequentTreeSum(TreeNode* root) {
        unordered_map<int,int> m;
        dfs(root,m);
    //map不能使用sort演算法，只能將其用vector儲存後，再使用sort進行排序
        vector<pair<int,int> > v;
        for(auto it:m){
            v.push_back(it);
        }
    //按照出現的頻率進行排序
        sort(v.begin(),v.end(),cmp);
        vector<int> res;
        if(v.empty()) return res;
        res.push_back(v[0].first);
        for(int i=1;i<v.size();i++){
            if(v[i].second!=v[0].second)
                break;
            res.push_back(v[i].first);
        }
        return res;
    }
};