设计getRandomPool结构

58 篇文章 1 订阅

【题目】设计一种结构,在该结构中有如下三个功能:
intsert(key):将某个key加入到该结构,做到不重复加入。
delete(key):将原本在结构中的某个key移除.
getRandom():等概率随机返回结构中的任何一个key。、
【要求】Insert、Delect和getRandom方法的时间复杂度都是O(1)

insert和getRandom都没什么好说的,delete方法要注意下不要直接删除,需要把最后一条填到要删除的位值,然后边界收缩一下,否则形成空洞会影响getRandom

class RandomPool{
    public:
        hash_map<string,long long int> map1;
        hash_map<long long int,string> map2;
        int size;
        RandomPool(){
            size=0;
        }

        void add(string str){
            map1.insert(pair<string,long long int>(str,size));
            map2.insert(pair<long long int,string>(size,str));
            size++;
        }
        string getRandom(){
            default_random_engine e;
            uniform_real_distribution<>u(0,size);
            int index=u(e);
            return map2[index];
        }
        void Delete(string str){
            if(map1.find(str)!=map1.end()){
                int  deleteIndex=map1[str];
                int lastIndex=--size;
                string lastKey=map2[lastIndex];
                map1.erase(str);
                map2.erase(deleteIndex);
                map1.insert(pair<string,long long int >(lastKey,deleteIndex));
                map2.insert(pair<long long int,string >(deleteIndex,lastKey));
            }
        }
    };
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值