380. 常数时间插入、删除和获取随机元素 动态数组+哈希表

题意:设计一个支持在平均时间复杂度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)动态数组和哈希表存储的元素个数

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值