力扣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); // 返回对应索引位置的元素

}

}

相关推荐
m0_587383002 小时前
广州 24 小时自助健身房解决方案技术实现与实战指南
java·spring·小程序·架构·需求分析
mmmmath_32 小时前
LeetCode.199.二叉树的右视图
数据结构·leetcode
·云扬·2 小时前
Prompt 工程与安全:从 System 设计到注入防御
java·人工智能·安全·ai·prompt
专注仿真2 小时前
引擎扩展性与集成设计深度剖析:从 14 个扩展点到 DIS→HLA 演进之路
开发语言·算法·仿真引擎·推演引擎
码爸3 小时前
支持向量机介绍与示例
算法·机器学习·支持向量机
怕浪猫3 小时前
2028编程圈很火的10个Skills
算法·面试·github
代码不停3 小时前
自动化测试(java)
java·开发语言·selenium·自动化
鬼手点金5 小时前
opencode-rules 插件
java·开发语言·前端·人工智能·chrome·git·github
paopaokaka_luck6 小时前
滑雪场管理系统(AI智能助手流式对话,ECharts 运营数据分析、教练、雪具与场地一体化预约,钱包支付退款、订单核销与服务评价)
java·人工智能·spring boot·网络协议·数据分析·echarts
Wang's Blog6 小时前
Java 项目实战: 交友项目 - Dubbo 的内部结构与一次调用的完整链路
java·dubbo