力扣面试150： O(1) 时间插入、删除和获取随机元素 HashMap结合数组

最新推荐文章于 2024-10-01 05:04:32 发布

兑生

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量413

点赞数 4

分类专栏： # 面试150 文章标签： leetcode 面试 java

本文链接：https://blog.csdn.net/lt6666678/article/details/137274741

版权

面试150 专栏收录该内容

41 篇文章 0 订阅

订阅专栏

本文介绍了一种使用哈希映射和数组实现的随机化集合类，支持O(1)时间复杂度的插入、删除和获取随机元素操作。通过维护元素值与数组下标的关系，实现在平均情况下高效的查找和修改操作。

摘要由CSDN通过智能技术生成

Problem: 380. O(1) 时间插入、删除和获取随机元素

思路

👩‍🏫 三叶题解

在这里插入图片描述

复杂度

时间复杂度: $O (1)$

空间复杂度: $O (n)$

Code

class RandomizedSet {
    static int[] nums = new int[200_010];//存元素的值
    Random random = new Random();
    Map<Integer, Integer> map = new HashMap<>();//key：元素的值，val：元素在nums数组中的下标
    int idx = -1;

    public boolean insert(int val) {
        if(map.containsKey(val))
            return false;
            nums[++idx] = val;
            map.put(val,idx);
            return true;
    }

    public boolean remove(int val) {
        if(!map.containsKey(val))
            return false;
        int cur = map.remove(val);//删除当前元素在 map 中的键值对，并返回 值（待删除元素在数组中的下标）
        if(cur != idx)//如果最后一个元素就是要删除的元素，直接跳过
            map.put(nums[idx],cur);//更新最后一个元素在数组中的下标
        nums[cur] = nums[idx--];//最后一个元素与当前元素交换，然后删除最后一个元素
        return true;
    }

    public int getRandom() {
        return nums[random.nextInt(idx+1)];
    }
}

/**
 * Your RandomizedSet object will be instantiated and called as such:
 * RandomizedSet obj = new RandomizedSet();
 * boolean param_1 = obj.insert(val);
 * boolean param_2 = obj.remove(val);
 * int param_3 = obj.getRandom();
 */