leetcode 381. O(1) 时间插入、删除和获取随机元素 - 允许重复 map

用一个map记录每一个元素出现的次数就可以
然后弄一个int记录总时间

这样 插入和删除的时候就将对应数字的mp值改掉就可以
但是获取随机元素就不是O(1)了 但是小于O(n)

class RandomizedCollection {
public:
    map<int ,int >mp;
    int cnt;
    /** Initialize your data structure here. */
    RandomizedCollection() {
        srand(time(0));
        cnt = 0;
    }

    /** Inserts a value to the collection. Returns true if the collection did not already contain the specified element. */
    bool insert(int val) {
        mp[val]++;
        cnt++;
        return true;
    }

    /** Removes a value from the collection. Returns true if the collection contained the specified element. */
    bool remove(int val) {
        --mp[val];
        if(mp[val]>=0){
            cnt--;
            return true;
        }else{
            mp[val] = 0;
            return false;
        }
    }

    /** Get a random element from the collection. */
    int getRandom() {
        int ran = rand()% cnt+1;
        map<int ,int >::iterator it = mp.begin();
        for(; it!= mp.end();it++){
            ran -= it->second;
            if(ran<=0){
                return it->first;
            }
        }
        return it->first;
    }
};

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值