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

}

}

相关推荐
列星随旋1 小时前
线段树和树状数组的学习
学习·算法
圣保罗的大教堂2 小时前
leetcode 61. 旋转链表 中等
leetcode
500842 小时前
昇腾 CANN 的五层架构,到底分了哪五层
java·人工智能·分布式·架构·ocr·wpf
摇滚侠2 小时前
Java 零基础全套教程,File 类与 IO 流,笔记 177-178
java·开发语言·笔记
全糖可乐气泡水3 小时前
Codex适配国产信创环境安装部署与技术适配全解析
开发语言·git·python·算法·百度
雨落在了我的手上3 小时前
初始java(十):类和对象(⼆)
java·开发语言
h_a_o777oah3 小时前
状态机+划分型 DP :深度解析K-划分问题下 DP 状态的转移逻辑(洛谷P2679 P2331 附C++代码)
c++·算法·动态规划·acm·状态机dp·划分型dp·滚动数组优化
05候补工程师3 小时前
从算法理想向工程现实的跨越:SLAM 核心架构、思维误区与 Nav2 实战避坑指南
人工智能·算法·安全·架构·机器人
莫雪歌4 小时前
Java AI 应用开发实践:基于 Spring Boot 实现 Chat、Memory、RAG 与 Tool Calling
java·aigc
SmartBrain4 小时前
AI全栈开发(SDD):慢病管理系统工程级设计
java·大数据·开发语言·人工智能·架构·aigc