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

题目描述

实现RandomizedSet 类：

RandomizedSet() 初始化 RandomizedSet 对象
bool insert(int val) 当元素 val 不存在时，向集合中插入该项，并返回 true ；否则，返回 false 。
bool remove(int val) 当元素 val 存在时，从集合中移除该项，并返回 true ；否则，返回 false 。
int getRandom() 随机返回现有集合中的一项（测试用例保证调用此方法时集合中至少存在一个元素）。每个元素应该有 相同的概率 被返回。
你必须实现类的所有函数，并满足每个函数的 平均 时间复杂度为 O(1) 。

示例：

输入
["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 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/insert-delete-getrandom-o1
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

由于题目要求插入删除和获取随机元素的时间复杂度都是常量级的，所以一定要用到一个哈希表，存储 {元素值：所在数组中的下标}的键值对，这种键值对的格式让我想到了python中的字典，用字典解决这题堪称完美。

插入元素时，查看被插入的元素是否已经存在，若存在，则插入失败，否则更新indices[val]=len(nums),并将其插入到nums末尾，这样就动态的维护了一个元素数组和一个记录元素所在位置的字典。

删除元素时，如果被删除的元素不存在，删除失败，否则找到该位置并删除

具体操作是用一个变量index记录删除元素所在的数组下标

index=indices[val]

并将数组中该下标对应元素更改为数组末尾元素，相当于是用末尾元素代替该元素，最后对该元素的删除操作可以转换为对末尾元素的删除，这样做的好处是在连续存储的数组中，不用因为删除中间元素而导致大量的移动后序元素。

具体操作为:

nums[index]=nums[-1],(python中数组下标-1表示最后一个元素)

indices[nums[index]]=index  (最后一个元素对应的位置更新为被删除元素所在位置)

nums.pop() (pop函数无参时，默认将末尾元素删除)

del indices[val] (最后将被删除元素之前存的索引键值对通过del关键字删除即可)

返回随机数组中的随机元素，刚好python中有一个对应的choice()函数，将数组即python中的列表作为该函数的参数传入，就可以随机的返回一个里面的元素，时间复杂度为O(1)

代码

class RandomizedSet:

    def __init__(self):
        self.nums=[]
        self.indices={}

    def insert(self, val: int) -> bool:
        if val in self.indices:
            return False
        self.indices[val]=len(self.nums)
        self.nums.append(val)
        return True

    def remove(self, val: int) -> bool:
        if val  not in self.indices:
            return False
        index=self.indices[val]
        self.nums[index]=self.nums[-1]
        self.indices[self.nums[index]]=index
        self.nums.pop()
        del self.indices[val]
        return True


    def getRandom(self) -> int:
        return choice(self.nums)