算法(TS):O(1) 时间插入、删除和获取随机元素

实现RandomizedSet 类:

  • RandomizedSet() 初始化 RandomizedSet 对象
  • bool insert(int val) 当元素 val 不存在时,向集合中插入该项,并返回 true ;否则,返回 false
  • bool remove(int val) 当元素 val 存在时,从集合中移除该项,并返回 true ;否则,返回 false
  • int getRandom() 随机返回现有集合中的一项(测试用例保证调用此方法时集合中至少存在一个元素)。每个元素应该有 相同的概率 被返回。

你必须实现类的所有函数,并满足每个函数的 平均 时间复杂度为 O(1)

解法

利用一个 hash 表保存元素的下标,使 remove 方法的时间复杂度为 O(1)

ts 复制代码
class RandomizedSet {
    nums: number[] = []
    numIndexHash: Map<number,number> = new Map()
    constructor() {}

    insert(val: number): boolean {
        if(this.numIndexHash.has(val)) {
            return false
        }

        const index = this.nums.length
        this.nums.push(val)
        this.numIndexHash.set(val,index)
        return true
    }

    remove(val: number): boolean {
        if(!this.numIndexHash.has(val)) {
            return false
        }

        const index = this.numIndexHash.get(val),
                lastIndex = this.nums.length-1,
                lastVal = this.nums[lastIndex]

        this.nums[index] = lastVal
        this.numIndexHash.set(lastVal,index)
        this.nums.pop()
        this.numIndexHash.delete(val)
        
        return true
    }

    getRandom(): number {
        const index = Math.floor(Math.random() * this.nums.length)
        return this.nums[index]
    }
}
相关推荐
CoovallyAIHub37 分钟前
智能“下沉”:边缘AI,更低功耗、更快响应、更强隐私,YOLO26只是开始
深度学习·算法·计算机视觉
Lear1 小时前
【链表】LeetCode 142.环形链表
算法
CoovallyAIHub1 小时前
2025目标检测模型全景图:从RF-DETR到YOLOv12,谁主沉浮?
深度学习·算法·计算机视觉
澪吟1 小时前
算法性能的核心度量:时间复杂度与空间复杂度全解析
数据结构·算法
咪咪渝粮2 小时前
108. 将有序数组转换为二叉搜索树
算法·leetcode
lzptouch2 小时前
蚁群(Ant Colony Optimization, ACO)算法
人工智能·算法·机器学习
苏纪云2 小时前
算法<C++>——双指针操作链表
c++·算法·链表·双指针
louisdlee.2 小时前
扫描线1:朴素扫描线
数据结构·c++·算法·扫描线
wan5555cn2 小时前
中国启用WPS格式进行国际交流:政策分析与影响评估
数据库·人工智能·笔记·深度学习·算法·wps
AndrewHZ2 小时前
【图像处理基石】图像形态学处理:从基础运算到工业级应用实践
图像处理·python·opencv·算法·计算机视觉·cv·形态学处理