Leetcode 501. 二叉搜索树中的众数 C++

Leetcode 501. 二叉搜索树中的众数

题目

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

假定 BST 有如下定义：

结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树

测试样例

给定 BST [1,null,2,2],

   1
    \
     2
    /
   2
返回[2].

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

题解

中序遍历，用哈希表记录数字出现的次数，同时求众数出现的次数，最后遍历哈希表找到众数。详细过程见代码

代码

	unordered_map<int,int> nums;
    int time=0;
    void search(TreeNode* root){
        if(root == NULL){
            return;
        }
        nums[root->val]++;
        time = max(time,nums[root->val]);
        search(root->left);
        search(root->right);
    }
    vector<int> findMode(TreeNode* root) {
        search(root);
        vector<int> ans;
        for(auto num : nums){
            if(num.second == time)
                ans.push_back(num.first);
        }
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-mode-in-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。