设计RandomPool结构

最新推荐文章于 2022-08-29 11:18:31 发布

BUPT-WT

最新推荐文章于 2022-08-29 11:18:31 发布

阅读量175

点赞数

分类专栏：秋招

本文链接：https://blog.csdn.net/weixin_41362649/article/details/94554709

版权

秋招专栏收录该内容

179 篇文章 2 订阅

订阅专栏

设计RandomPool结构
【题目】设计一种结构,在该结构中有如下三个功能: insert(key):将某个key加入到该结构,做到不重复加入。 delete(key):将原本在结构中的某个key移除。 getRandom(): 等概率随机

返回结构中的任何一个key。
【要求】 Insert、delete和getRandom方法的时间复杂度都是 O(1)

思路：对每个元素都加上一个下标，第一个元素的下标为0，每加入一个元素，对应的下标加1。如果要删除一个元素，假设该元素的下标为i，将最后的一个元素放到位置i处，同时删除最后一个元素（对应的下标也删除）。这样处理就保证了下标的连续性，在进行getRandom操作的时候，只需要随机产生一个下标，将该下标位置处的元素返回即可。为了得到key -> index以及index -> key的对应关系，使用两个哈希表来建立两者之间的映射。

class pool:
    def __init__(self):
        self.indexMapKey = {}
        self.keyMapindex = {}
        self.size = 0

    def insert(self,value):
        self.indexMapKey[self.size] = value
        self.keyMapindex[value] = self.size
        self.size +=1

    def delete(self,key):
        index = self.keyMapindex[key]
        lastKey = self.indexMapKey[self.size-1]
        
        self.indexMapKey[index] = lastKey
        self.keyMapindex[lastKey] = index
        
        del self.keyMapindex[lastKey]
        del self.indexMapKey[self.size-1]

        self.size -=1

    def getRandom(self):

        if self.index == 0:
            return

        index = int(random.random()*self.size)
        
        return self.indexMapKey[index]