题意:设计一个支持在平均时间复杂度O(1)下,执行以下操作的数据结构,insert(val),remove(val),getRandom
我们具有两个平均插入时间为O(1)的选择,哈希表和动态数组list
虽然哈希表提供常数时间的插入和删除,但是getrandom需要使用一个索引返回一个元素,而哈希表中没有索引,需要将哈希表的键转为列表,而这需要线性时间,所以使用列表list存储值。列表list可以实现常数时间的insert和getRandom,但如何remove,删除任意索引元素需要线性时间,这里的解决方案总是删除最后一个元素。
将要删除元素和最后一个元素交换,将最后一个元素删除。为此,必须在常数时间获取要删除元素的索引,因此需要一个哈希表来存储值到索引的映射
动态数组存储元素值,哈希表存储值到索引的映射
from random import choice
class RandomizedSet():
def __init__(self):
self.dict={}
self.list=[]
def insert(self,val):
if val in self.dict:
return False
self.dict[val]=len(self.list)
self.list.append(val)
return True
def remove(self,val):
if val in self.dict:
index , last_element=self.dict[val],self.list[-1]#相当于将最后一个值移到index位置
self.list[index],self.dict[last_element]=last_element,index
self.list.pop()
del self.dict[val]
return True
return False
def getRandom(self):
return choice(self.list)
时间复杂度O(1)空间复杂度O(n)动态数组和哈希表存储的元素个数