【NO.12】LeetCode经典150题-380. O(1) 时间插入、删除和获取随机元素

文章目录

  • [【NO.12】LeetCode经典150题-380. O(1) 时间插入、删除和获取随机元素](#【NO.12】LeetCode经典150题-380. O(1) 时间插入、删除和获取随机元素)
  • 解题

【NO.12】LeetCode经典150题-380. O(1) 时间插入、删除和获取随机元素

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 。

提示:

  • -231 <= val <= 231 - 1
  • 最多调用 insertremovegetRandom 函数 2 * 10^5
  • 在调用 getRandom 方法时,数据结构中 至少存在一个 元素。

解题

要求实现函数的平均时间复杂度都是O(1),那么:

  • 获取需要列表,用下标来操作,可以达到要求
  • 插入,列表的尾部插入,平均时间复杂度是O(1)
  • 删除,可以用哈希表来保存元素对应的下标,然后找到下标后,操作列表
java 复制代码
class RandomizedSet {

    // 数组,用于获取元素
    List<Integer> nums;
    // map.key为数,value为下标,用于确定下标位置
    Map<Integer, Integer> map;
    Random random;

    public RandomizedSet() {
        nums = new ArrayList<Integer>();
        map = new HashMap<Integer, Integer>();
        random = new Random();
    }
    
    // insert时,数组列表与map都要维护
    public boolean insert(int val) {
        if (!map.containsKey(val)) {
            int length = nums.size();
            nums.add(val);
            map.put(val, length);
            return true;
        }
        return false;
    }
    
    // 先找下标
    // 用列表最后一个数覆盖要删除的数,并更新map
    // 删除最后一个数
    public boolean remove(int val) {
        if (map.containsKey(val)) {
            int length = nums.size();
            Integer lastNum = nums.get(length - 1);
            int targetIndex = map.get(val);
            // 用最后一个数覆盖要删除的数
            nums.set(targetIndex, lastNum);
            map.put(lastNum, targetIndex);
            // 删除最后一个数
            nums.remove(length - 1);
            map.remove(val);
            return true;
        }
        return false;
    }
    
    public int getRandom() {
        int index = random.nextInt(nums.size());
        return nums.get(index);
    }
}

/**
 * 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();
 */
相关推荐
未来之窗软件服务1 小时前
自己写算法(九)网页数字动画函数——东方仙盟化神期
前端·javascript·算法·仙盟创梦ide·东方仙盟·东方仙盟算法
豐儀麟阁贵1 小时前
基本数据类型
java·算法
乐迪信息3 小时前
乐迪信息:基于AI算法的煤矿作业人员安全规范智能监测与预警系统
大数据·人工智能·算法·安全·视觉检测·推荐算法
hsjkdhs3 小时前
C++之多层继承、多源继承、菱形继承
开发语言·c++·算法
立志成为大牛的小牛4 小时前
数据结构——十七、线索二叉树找前驱与后继(王道408)
数据结构·笔记·学习·程序人生·考研·算法
星空下的曙光4 小时前
Node.js crypto模块所有 API 详解 + 常用 API + 使用场景
算法·node.js·哈希算法
StarPrayers.6 小时前
旅行商问题(TSP)(2)(heuristics.py)(TSP 的两种贪心启发式算法实现)
前端·人工智能·python·算法·pycharm·启发式算法
爱吃橘的橘猫6 小时前
嵌入式系统与嵌入式 C 语言(2)
c语言·算法·嵌入式
235166 小时前
【LeetCode】146. LRU 缓存
java·后端·算法·leetcode·链表·缓存·职场和发展
weixin_307779137 小时前
使用Python高效读取ZIP压缩文件中的UTF-8 JSON数据到Pandas和PySpark DataFrame
开发语言·python·算法·自动化·json