O(1) 时间插入、删除和获取随机元素(Java实现)

设计一个数据结构,支持在平均时间复杂度O(1)下进行插入、删除和随机获取元素的操作。采用动态数组配合哈希表实现,哈希表用于记录元素在数组中的所有位置,删除操作通过交换待删除元素与数组末尾元素并更新哈希表来完成,随机获取元素则通过随机生成数组下标来达到线性概率效果。
摘要由CSDN通过智能技术生成

设计一个支持在平均 时间复杂度 O(1) 下, 执行以下操作的数据结构。

注意: 允许出现重复元素。

insert(val):向集合中插入元素 val。
remove(val):当 val 存在时,从集合中移除一个 val。
getRandom:从现有集合中随机获取一个元素。每个元素被返回的概率应该与其在集合中的数量呈线性相关。

题目要求我们在时间复杂度在O(1)下执行上述所有操作,首先我们需要选择一个集合来存放数据,此处我们的选择是动态数组,对于动态数组来说,其在末尾插入元素的时间复杂度能够满足要求,就是O(1)。

对于移除操作来说,为了实现时间复杂度为O(1),我们需要用到一个哈希表,存储一个元素在集合中的所有下标,当需要删除该元素时,将该元素与集合的末尾元素交换位置,同时更新交换后末尾元素的下标变化,最后再删除集合的末尾元素。

对于随机获取一个元素,为了实现每个元素被返回的概率应该与其在集合中的数量呈线性相关,我们可以考虑设计随机函数得到集合范围内的任意一个下标,然后返回集合中该下标对应的元素。

class RandomizedCollection {
   
    Map<Integer, Set<Integer>> index;
    List<Integer> nums;

    /** 初始化集合 */
    public RandomizedCollection() {
   
        index = new HashMap<Integer, Set<Integer>>();
        nums = new ArrayList<Integer
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值