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

// 定义一个RandomizedSet类,用于实现随机集合的功能

class RandomizedSet {

// nums列表用于存储集合中的元素

List<Integer> nums;

// indices映射用于存储元素值到其在nums列表中的索引的映射

Map<Integer, Integer> indices;

// random对象用于生成随机数

Random random;

// 构造函数,初始化nums、indices和random

public RandomizedSet() {

nums = new ArrayList<Integer>(); // 初始化nums为空列表

indices = new HashMap<Integer, Integer>(); // 初始化indices为空映射

random = new Random(); // 初始化random对象

}

// 插入元素的方法

public boolean insert(int val) {

if (indices.containsKey(val)) { // 如果元素已存在,返回false

return false;

}

int index= nums.size(); // 获取当前nums的大小作为新元素的索引

nums.add(val); // 将元素添加到nums列表末尾

indices.put(val, index); // 更新indices映射,将元素值映射到其索引

return true; // 插入成功,返回true

}

// 删除元素的方法

public boolean remove(int val) {

if (!indices.containsKey(val)) { // 如果元素不存在,返回false

return false;

}

int index = indices.get(val); // 获取要删除元素的索引

int last = nums.get(nums.size() - 1); // 获取nums列表最后一个元素

nums.set(index, last); // 将最后一个元素放到要删除元素的位置

indices.put(last, index); // 更新indices映射,将最后一个元素的索引映射到被删除元素的位置

nums.remove(nums.size() - 1); // 从nums列表中移除最后一个元素

indices.remove(val); // 从indices映射中移除被删除元素的映射关系

return true; // 删除成功,返回true

}

// 获取随机元素的方法

public int getRandom() {

int randomIndex = random.nextInt(nums.size()); // 生成一个随机索引

return nums.get(randomIndex); // 返回对应索引位置的元素

}

}

相关推荐
To_OC5 小时前
LC 51 N 皇后:我以为难的是回溯,结果栽在了对角线下标
javascript·算法·leetcode
腻害兔5 小时前
【若依项目-产品经理视角】深度拆解 RuoYi-Vue-Pro 商城模块:从商品管理到交易引擎,50 张表撑起一整套电商系统
java·大数据·vue.js·产品经理·ai编程
2401_841495646 小时前
【操作系统】进程同步与互斥实验报告
c++·算法·操作系统·进程·并发·同步·互斥
码智社6 小时前
AES加密原理详解及Java实现加解密实战
java·开发语言
萧瑟余晖7 小时前
JDK 26 新特性详解
java·开发语言
马优晨8 小时前
Freemarker 完整讲解(后端 Java 模板引擎)
java·开发语言·freemarker·freemarker 完整讲解·freemarker模板引擎
爱刷碗的苏泓舒10 小时前
PPP-AR 中的参考星选取:数学原理、评价指标与切换处理
算法·gnss·模糊度固定·ppp-ar·星间单差·参考星·卫星端偏差
维天说10 小时前
CLI-Switch 2026年3月版历史设计:Hook、TTY 隔离与 JSON 状态
java·服务器·json
Zane199411 小时前
并发 vs 并行:别再傻傻分不清了,一文讲透 Java 并发编程的第一课
java·后端