設計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 = 0def insert(self,value):self.indexMapKey[self.size] = valueself.keyMapindex[value] = self.sizeself.size +=1def delete(self,key):index = self.keyMapindex[key]lastKey = self.indexMapKey[self.size-1]self.indexMapKey[index] = lastKeyself.keyMapindex[lastKey] = indexdel self.keyMapindex[lastKey]del self.indexMapKey[self.size-1]self.size -=1def getRandom(self):if self.index == 0:returnindex = int(random.random()*self.size)return self.indexMapKey[index]

總結

以上是生活随笔為你收集整理的设计RandomPool结构的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。