O(1) 时间插入、删除和获取随机元素（2022-4-13）每日一练

380. O(1) 时间插入、删除和获取随机元素（2022-4-13）

实现RandomizedSet 类：

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

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

详细示例请移步官网查看

解题思路

既然都说设计一个集合的数据结构，那就直接使用js自带的set就好了；考虑一下插入、删除、获取三个操作。

插入：set.add() 直接可以插入，且set.has()也可以查询是否已经存在。

删除：set.delete()直接可以删除。

获取随机数： Math.random()乘上集合长度获取随机数，然后[...set][randomIndex]扩展set为数组直接通过下标获取值。

class RandomizedSet {
  constructor(){
    this.set = new Set()
    this.nums = 0
  }；
  insert = function(val) {
    if(this.set.has(val)) return false
    this.set.add(val)
    this.nums += 1
    return true
  };

  remove = function(val) {
    if(!this.set.has(val))  return false
    this.set.delete(val)
    this.nums -= 1
    return true
  };

  getRandom = function() {
    let lucky = Math.floor(Math.random() * this.nums)
    return [...this.set][lucky]
  };
};

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