【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】【PyTorch】【强化学习简介优化框架】
算法
loser~曹6 分钟前
基于快速排序解决 leetcode hot215 查找数组中第k大的数字
数据结构·算法·leetcode
Dream it possible!12 分钟前
LeetCode 热题 100_打家劫舍(83_198_中等_C++)(动态规划)
c++·算法·leetcode·动态规划
zhouziyi070117 分钟前
【蓝桥杯14天冲刺课题单】Day 8
c++·算法·蓝桥杯
SylviaW0821 分钟前
python-leetcode 62.搜索插入位置
数据结构·算法·leetcode
舔甜歌姬的EGUMI LEGACY2 小时前
【算法day28】解数独——编写一个程序,通过填充空格来解决数独问题
算法
welkin2 小时前
KMP 个人理解
前端·算法
半桔2 小时前
红黑树剖析
c语言·开发语言·数据结构·c++·后端·算法
eason_fan2 小时前
前端面试手撕代码(字节)
前端·算法·面试
今天_也很困2 小时前
牛客2025年愚人节比赛
c++·算法