JavaScript-O(1) 时间插入、删除和获取随机元素(实现一个类)

本文链接：https://blog.csdn.net/qq_54140719/article/details/133815326

一、题目描述

实现RandomizedSet 类：

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

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

二、解题思路

1.题目要求平均时间复杂度为O(1)。如果使用数组来插入，删除，必然会遍历数组查询是否存在，时间复杂度为O(n)。所以我们可以使用map结构，将存入的val值作为key。判断是否已经存在，就看key是否存在，使用has(key)判断。

2.插入、删除解决了，但是随机返回集合中的一项，map却不能实现。（因为存入的key不一定是连续的，获取到随机的随机数不一定可以找到对应的key），所以我们需要一个数组来辅助，获取随机的下标就可以提取出随机的值。

3.存入时(存入val)：将val值作为key，在数组中的索引作为value。删除时(删除val)：在map结构中，通过val(key)很快找到索引。通过索引删除数组的值(如果删除的是中间某个数，那后面值的位置会发生变化)，为保证其他值的位置(索引不变)，可以将最后一个值赋给当前这个值，完成覆盖，这样只变动了最后一个值的位置，其他值的位置不会变，只会多出(重复)最后一个值，使用pop()即可，然后在map中删除val，并更新被修改位置的值即可，完成map与数组的统一。

三、解题代码

示例一：


var RandomizedSet = function() {
    this.obj = new Map()
    this.arr = []
};

/** 
 * @param {number} val
 * @return {boolean}
 */
RandomizedSet.prototype.insert = function(val) {
    if(!this.obj.has(val)) {    
        // 将下标作为值存入
        this.obj.set(val,this.arr.length)
        this.arr[this.arr.length] = val
        return true
    }
    return false
};

/** 
 * @param {number} val
 * @return {boolean}
 */
RandomizedSet.prototype.remove = function(val) {
    if(this.obj.has(val)) {
        // 获取val对应的下标
        let index = this.obj.get(val)
        //用最后一个元素覆盖掉当前元素，起到删除的作用
        this.arr[index] = this.arr[this.arr.length-1]
        //将最后一个元素出栈
        this.arr.pop()
        //在map中，更新最后一个元素，他的值为当前索引
        this.obj.set(this.arr[index],index)
        this.obj.delete(val)
        return true
    }
    return false   
};

/**
 * @return {number}
 */
RandomizedSet.prototype.getRandom = function() {
    if(this.arr.length === 0) return []
    let index = Math.floor(Math.random()*this.arr.length) 
    return this.arr[index]
};

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

代码解析：代码的关键点在删除元素这里：先使用 this.obj.has(val) 判断被删除 val是否存在，如果存在，先通过get() 获取在数组中的索引(let index = this.obj.get(val)，然后this.arr[index] = this.arr[this.arr.length-1] 用最后一个元素覆盖掉当前元素，起到删除的作用。使用pop()将最后多出的元素去掉。数组最后一个元素目前已经到index这个位置，this.obj.set(this.arr[index],index) 更新在map中的数据，最后this.obj.delete(val)，将val删除。