【C++】每日一题 380 O(1)时间插入，删除和获取随机元素

最新推荐文章于 2024-12-14 17:35:48 发布

原创

最新推荐文章于 2024-12-14 17:35:48 发布 · 525 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #开发语言

实现RandomizedSet 类：

RandomizedSet() 初始化 RandomizedSet 对象
bool insert(int val) 当元素 val 不存在时，向集合中插入该项，并返回 true ；否则，返回 false 。
bool remove(int val) 当元素 val 存在时，从集合中移除该项，并返回 true ；否则，返回 false 。
int getRandom() 随机返回现有集合中的一项（测试用例保证调用此方法时集合中至少存在一个元素）。每个元素应该有相同的概率被返回。
你必须实现类的所有函数，并满足每个函数的平均时间复杂度为 O(1) 。

#include <unordered_map>
#include <vector>
#include <cstdlib>

class RandomizedSet {
   
   
private:
    std::unordered_map<int, int> valToIndex; // 用于存储值到索引的映射
    std::vector<int> values; // 用于存储集合中的值

public:
    /** Initialize your data structure here. */
    RandomizedSet() {
   
   
        
    }