前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >380. O(1) 时间插入、删除和获取随机元素

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

作者头像
狼啸风云
发布2023-10-07 16:09:26
1480
发布2023-10-07 16:09:26
举报
文章被收录于专栏:计算机视觉理论及其实现

实现RandomizedSet 类:

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

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

示例:

代码语言:javascript
复制
输入
["RandomizedSet", "insert", "remove", "insert", "getRandom", "remove", "insert", "getRandom"]
[[], [1], [2], [2], [], [1], [2], []]
输出
[null, true, false, true, 2, true, false, 2]

解释
RandomizedSet randomizedSet = new RandomizedSet();
randomizedSet.insert(1); // 向集合中插入 1 。返回 true 表示 1 被成功地插入。
randomizedSet.remove(2); // 返回 false ,表示集合中不存在 2 。
randomizedSet.insert(2); // 向集合中插入 2 。返回 true 。集合现在包含 [1,2] 。
randomizedSet.getRandom(); // getRandom 应随机返回 1 或 2 。
randomizedSet.remove(1); // 从集合中移除 1 ,返回 true 。集合现在包含 [2] 。
randomizedSet.insert(2); // 2 已在集合中,所以返回 false 。
randomizedSet.getRandom(); // 由于 2 是集合中唯一的数字,getRandom 总是返回 2 。

方法一:变长数组 + 哈希表

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

O(1)
O(1)

。变长数组可以在

O(1)
O(1)

的时间内完成获取随机元素操作,但是由于无法在

O(1)
O(1)

的时间内判断元素是否存在,因此不能在

O(1)
O(1)

的时间内完成插入和删除操作。哈希表可以在

O(1)
O(1)

的时间内完成插入和删除操作,但是由于无法根据下标定位到特定元素,因此不能在

O(1)
O(1)

的时间内完成获取随机元素操作。为了满足插入、删除和获取随机元素操作的时间复杂度都是

O(1)
O(1)

,需要将变长数组和哈希表结合,变长数组中存储元素,哈希表中存储每个元素在变长数组中的下标。

插入操作时,首先判断

\textit{val}
\textit{val}

是否在哈希表中,如果已经存在则返回

\text{false}
\text{false}

,如果不存在则插入

\textit{val}
\textit{val}

,操作如下:

    在变长数组的末尾添加

\textit{val}
\textit{val}

    在添加

\textit{val}
\textit{val}

之前的变长数组长度为

\textit{val}
\textit{val}

所在下标

\textit{index}
\textit{index}

,将 \textit{val} 和下标

\textit{index}
\textit{index}

存入哈希表;

    返回

\text{true}
\text{true}

删除操作时,首先判断

\textit{val}
\textit{val}

是否在哈希表中,如果不存在则返回

\text{false}
\text{false}

,如果存在则删除

\textit{val}
\textit{val}

,操作如下:

    从哈希表中获得

\textit{val}
\textit{val}

的下标

\textit{index}
\textit{index}

    将变长数组的最后一个元素

\textit{last}
\textit{last}

移动到下标

\textit{index}
\textit{index}

处,在哈希表中将

\textit{last}
\textit{last}

的下标更新为

\textit{index}
\textit{index}

    在变长数组中删除最后一个元素,在哈希表中删除

\textit{val}
\textit{val}

    返回

\text{true}
\text{true}

删除操作的重点在于将变长数组的最后一个元素移动到待删除元素的下标处,然后删除变长数组的最后一个元素。该操作的时间复杂度是

O(1)
O(1)

,且可以保证在删除操作之后变长数组中的所有元素的下标都连续,方便插入操作和获取随机元素操作。

获取随机元素操作时,由于变长数组中的所有元素的下标都连续,因此随机选取一个下标,返回变长数组中该下标处的元素。

代码语言:javascript
复制
class RandomizedSet {
public:
    RandomizedSet() {
        srand((unsigned)time(NULL));
    }
    
    bool insert(int val) {
        if (indices.count(val)) {
            return false;
        }
        int index = nums.size();
        nums.emplace_back(val);
        indices[val] = index;
        return true;
    }
    
    bool remove(int val) {
        if (!indices.count(val)) {
            return false;
        }
        int index = indices[val];
        int last = nums.back();
        nums[index] = last;
        indices[last] = index;
        nums.pop_back();
        indices.erase(val);
        return true;
    }
    
    int getRandom() {
        int randomIndex = rand()%nums.size();
        return nums[randomIndex];
    }
private:
    vector<int> nums;
    unordered_map<int, int> indices;
};
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
对象存储
对象存储(Cloud Object Storage,COS)是由腾讯云推出的无目录层次结构、无数据格式限制,可容纳海量数据且支持 HTTP/HTTPS 协议访问的分布式存储服务。腾讯云 COS 的存储桶空间无容量上限,无需分区管理,适用于 CDN 数据分发、数据万象处理或大数据计算与分析的数据湖等多种场景。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档