Leetcode501-二叉搜索树中的众数

Leetcode501-二叉搜索树中的众数(C++)

给定一个有相同值的二叉搜索树（BST），找出 BST 中的所有众数（出现频率最高的元素）。

假定 BST 有如下定义：

结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树
例如：
给定 BST [1,null,2,2],

1

2
/
2
返回[2].

提示：如果众数超过1个，不需考虑输出顺序

思路：对整个二叉树进行深搜，用散列表记录每个元素出现的次数，然后把出现次数同为最大的元素插入到数组里面进行输出，主要想记录的是迭代器里面first和second，分别指向第一维和第二维的值。用unordered_map是因为比map容器能更快地通过它们的键访问单个元素。

代码：

class Solution {
public:
    vector<int> findMode(TreeNode* root) {
        if(root == NULL)
        return {};
        cnt[root->val]++;
        dfs(root->left);
        dfs(root->right);
        int max = 0;
        vector<int>re;
        for(auto it = cnt.begin(); it != cnt.end(); it++)
        max = it->second > max ? it->second : max;
        for(auto ip = cnt.begin(); ip != cnt.end(); ip++)
        if(ip->second == max)
        re.push_back(ip->first);
        return re;
    }
    unordered_map<int,int>cnt;
    void dfs(TreeNode *root)
    {
        if(root == NULL)
        return;
        cnt[root->val]++;
        dfs(root->left);
        dfs(root->right);
        return;
    }
};