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

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

假定 BST 有如下定义:

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

例如:
给定 BST [1,null,2,2],

   1
    \
     2
    /
   2

返回[2].

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

进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)

【分析】这里的众数就是出现最多次的数字,可以有多个,那么这道题比较直接点思路就是利用一个哈希表来记录数字和其出现次数之前的映射,然后维护一个变量max_count来记录当前最多的次数值,这样在遍历完树之后,根据这个max_count值就能把对应的元素找出来。那么用这种方法的话就不需要用到二分搜索树的性质了,随意一种遍历方式都可以,下面我们来看迭代的中序遍历的解法如下:

class Solution {
public:
    vector<int> findMode(TreeNode* root) {
        if(!root) return {};
        TreeNode* p = root;
        vector<int> res;
        stack<TreeNode*> s;
        int max_count = 0;
        unordered_map<int,int> m;
        while( p || !s.empty() ){
            while(p){
                s.push(p);
                p = p->left;
            }
            p = s.top();s.pop();
            max_count = max(max_count,++m[ p->val ]);
            p = p->right;
        }
        for(auto c : m){
            if(c.second == max_count){
                res.push_back(c.first);
            }
        }
        return res; 
    }
};

 注意中序遍历的顺序是沿着最左的分支遍历到头,然后遍历最左节点的右子树,再然后才回溯到上层节点。这道题很好的练习了中序遍历的迭代实现。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值