【leetcode--O(1) 时间插入、删除和获取随机元素】

这道题要求实现一个类,满足插入、删除和获取随机元素操作的平均时间复杂度为 O(1)。

变长数组可以在 O(1) 的时间内完成获取随机元素操作 ,但是由于无法在 O(1)的时间内判断元素是否存在,因此不能在 O(1) 的时间内完成插入和删除操作。哈希表可以在 O(1) 的时间内完成插入和删除操作, 但是由于无法根据下标定位到特定元素,因此不能在 O(1) 的时间内完成获取随机元素操作。为了满足插入、删除和获取随机元素操作的时间复杂度都是 O(1),需要将变长数组和哈希表结合,变长数组中存储元素,哈希表中存储每个元素在变长数组中的下标。

总括:数组提供快速访问,哈希表提供快速查找。

首先创建两个实例化对象

  1. self.nums 是一个列表,用于存储插入的数字。
  2. self.indices 是一个字典,用于存储插入的数字以及其在 self.nums 列表中的索引。

创建RandomizedSet类的实例时,会自动创建一个空的nums列表和一个空的indices字典,为后续插入删除和随机获取做准备。

choice 是 Python 标准库中的 random 模块中的函数,用于从列表中随机选择一个元素并返回。

比较难理解的是remove这个类,过程:

首先检查给定的值是否存在indices字典中,如果不在,就返回False,表示删除失败

如果值存在,他会取出该值在nums列表中的索引id,然后他将nums列表中索引为id的元素替换为nums列表中的最后一个元素的值,这样就删除了原始的val,更新indices字典最后一个元素的索引值为id。删除nums列表中的最后一个元素,然后从indices列表中删除给定值。

这如果还不理解的话,可以自己画一个图就理解了。

例如nums:2,1,3

indices:(2:0),(1:1),(3:2)

举例:

(1)insert,插入4:

在字典里增加值为"4"的点indices[4]的值为len(nums)= 3, nums增加4。

(2)remove。删除1:

获取"1"的值(序号),id为1。在nums找到序号为1的元素,将值更改为最后一个元素的值,即3.

变为:nums:2,3,3

indices:(2:0),(1:1),(3:2)

nums[1]值为1,indice里面"3"对应的值改为id=1

变为:nums:2,3,3

indices:(2:0),(1:1),(3:1)

然后弹出nums最后一个元素,将indices中元素"1"删除。

变为:nums:2,3

indices:(2:0),(3:1)

复制代码
class RandomizedSet:
    def __init__(self):
        self.nums = []
        self.indices = {}

    def insert(self, val: int) -> bool:
        if val in self.indices:
            return False
        self.indices[val] = len(self.nums)
        self.nums.append(val)
        return True

    def remove(self, val: int) -> bool:
        if val not in self.indices:
            return False
        id = self.indices[val]
        self.nums[id] = self.nums[-1]
        self.indices[self.nums[id]] = id
        self.nums.pop()
        del self.indices[val]
        return True

    def getRandom(self) -> int:
        return choice(self.nums)
相关推荐
alphaTao1 分钟前
LeetCode 每日一题 2025/11/3-2025/11/9
windows·leetcode
2301_796512524 分钟前
Rust编程学习 - 问号运算符会return一个Result 类型,但是如何使用main函数中使用问号运算符
开发语言·学习·算法·rust
小龙报20 分钟前
算法通关指南:数据结构和算法篇 --- 队列相关算法题》--- 1. 【模板】队列,2. 机器翻译
c语言·开发语言·数据结构·c++·算法·学习方法·visual studio
晨非辰37 分钟前
【数据结构初阶】--从排序算法原理分析到代码实现操作,参透插入排序的奥秘!
c语言·开发语言·数据结构·c++·算法·面试·排序算法
三川6982 小时前
排序算法介绍
数据结构·算法·排序算法
智驱力人工智能6 小时前
基于视觉分析的人脸联动使用手机检测系统 智能安全管理新突破 人脸与手机行为联动检测 多模态融合人脸与手机行为分析模型
算法·安全·目标检测·计算机视觉·智能手机·视觉检测·边缘计算
2301_764441337 小时前
水星热演化核幔耦合数值模拟
python·算法·数学建模
循环过三天7 小时前
3.4、Python-集合
开发语言·笔记·python·学习·算法
priority_key9 小时前
排序算法:堆排序、快速排序、归并排序
java·后端·算法·排序算法·归并排序·堆排序·快速排序
不染尘.10 小时前
2025_11_7_刷题
开发语言·c++·vscode·算法