380. O(1) 时间插入、删除和获取随机元素 : 数据结构运用题

原创

宫水三叶 2022-05-20 15:03:55 ©著作权

文章标签 后端 Java 算法数组数据结构 文章分类 数据结构与算法人工智能

©著作权归作者所有：来自51CTO博客作者宫水三叶的原创作品，请联系作者获取转载授权，否则将追究法律责任

题目描述

这是 LeetCode 上的 380. O(1) 时间插入、删除和获取随机元素 ，难度为中等。

Tag : 「数据结构」、「哈希表」

实现 RandomizedSet 类：

RandomizedSet() 初始化RandomizedSet 对象
bool insert(int val) 当元素val 不存在时，向集合中插入该项，并返回true；否则，返回false。
bool remove(int val) 当元素val 存在时，从集合中移除该项，并返回true；否则，返回false。
int getRandom() 随机返回现有集合中的一项（测试用例保证调用此方法时集合中至少存在一个元素）。每个元素应该有相同的概率被返回。

你必须实现类的所有函数，并满足每个函数的平均时间复杂度为。

示例：

输入
["RandomizedSet", "insert", "remove", "insert", "getRandom", "remove", "insert", "getRandom"]
[[], [1], [2], [2], [], [1], [2], []]

输出
[null, true, false, true, 2, true, false, 2]

解释
RandomizedSet randomizedSet = new RandomizedSet();
randomizedSet.insert(1); // 向集合中插入 1 。返回 true 表示 1 被成功地插入。
randomizedSet.remove(2); // 返回 false ，表示集合中不存在 2 。
randomizedSet.insert(2); // 向集合中插入 2 。返回 true 。集合现在包含 [1,2] 。
randomizedSet.getRandom(); // getRandom 应随机返回 1 或 2 。
randomizedSet.remove(1); // 从集合中移除 1 ，返回 true 。集合现在包含 [2] 。
randomizedSet.insert(2); // 2 已在集合中，所以返回 false 。
randomizedSet.getRandom(); // 由于 2 是集合中唯一的数字，getRandom 总是返回 2 。

提示：

最多调用insert、remove 和getRandom 函数次
在调用getRandom 方法时，数据结构中至少存在一个元素。

哈希表 + 删除交换

对于 insert 和 remove 操作容易想到使用「哈希表」来实现复杂度，但对于 getRandom 操作，比较理想的情况是能够在一个数组内随机下标进行返回。

将两者结合，我们可以将哈希表设计为：以入参 val 为键，数组下标 loc 为值。

为了确保严格，我们不能「使用拒绝采样」和「在数组非结尾位置添加/删除元素」。

因此我们需要申请一个足够大的数组 nums（利用数据范围为），并使用变量 idx 记录当前使用到哪一位（即下标在范围内均是存活值）。

对于几类操作逻辑：

insert 操作：使用哈希表判断val 是否存在，存在的话返回fasle，否则将其添加到nums，更新idx，同时更新哈希表；
remove 操作：使用哈希表判断val 是否存在，不存在的话返回false，否则从哈希表中将val 删除，同时取出其所在nums 的下标loc，然后将nums[idx] 赋值到loc 位置，并更新idx（含义为将原本处于loc 位置的元素删除），同时更新原本位于idx 位置的数在哈希表中的值为loc（若loc 与idx 相等，说明删除的是最后一个元素，这一步可跳过）；
getRandom 操作：由于我们人为确保了均为存活值，因此直接在范围内进行随机即可。

代码：

class RandomizedSet {
    static int[] nums = new int[200010];
    Random random = new Random();
    Map<Integer, Integer> map = new HashMap<>();
    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 loc = map.remove(val);
        if (loc != idx) map.put(nums[idx], loc);
        nums[loc] = nums[idx--];
        return true;
    }
    public int getRandom() {
        return nums[random.nextInt(idx + 1)];
    }
}