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

实现RandomizedSet 类:

RandomizedSet() 初始化 RandomizedSet 对象

bool insert(int val) 当元素 val 不存在时,向集合中插入该项,并返回 true ;否则,返回 false 。

bool remove(int val) 当元素 val 存在时,从集合中移除该项,并返回 true ;否则,返回 false 。

int getRandom() 随机返回现有集合中的一项(测试用例保证调用此方法时集合中至少存在一个元素)。每个元素应该有 相同的概率 被返回。

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

字节面试遇到过的题

考虑删除的时候把最后一个数换到当前要删除的位置

cpp 复制代码
class RandomizedSet {
	vector<int>v;
	unordered_map<int,int>mp1;
public:
	
    RandomizedSet() {
        v.clear();
        mp1.clear();
    }
    
    bool insert(int val) {
        if(mp1.find(val)==mp1.end())
        {
			v.push_back(val);
			mp1[val]=v.size()-1;
			return true;
		}
		return false;
    }
    
    bool remove(int val) {
        if(mp1.find(val)!=mp1.end())
        {
        	int end_num=v.back();
        	v.pop_back();
        	int pos=mp1[val];
        	v[pos]=end_num;
        	mp1[end_num]=pos;
        	mp1.erase(val);
			return true;
		}
		return false;
    }
    
    int getRandom() {
        int pos=rand()%v.size();
        return v[pos];
    }
};
相关推荐
lifallen27 分钟前
深入解析RocksDB的MVCC和LSM Tree level
大数据·数据结构·数据库·c++·lsm-tree·lsm tree
金融小师妹1 小时前
AI量化模型解析黄金3300关口博弈:市场聚焦“非农数据”的GRU-RNN混合架构推演
大数据·人工智能·算法
金融小师妹1 小时前
基于LSTM-GRU混合网络的动态解析:美联储维稳政策与黄金单日跌1.5%的非线性关联
大数据·人工智能·算法
白日梦想家-K1 小时前
题单【模拟与高精度】
开发语言·c++·算法
岁忧2 小时前
(LeetCode 面试经典 150 题) 138. 随机链表的复制 (哈希表)
java·c++·leetcode·链表·面试·go
重生之我是Java开发战士2 小时前
【C语言】内存函数与数据在内存中的存储
c语言·开发语言·算法
roman_日积跬步-终至千里3 小时前
【机器学习】“回归“算法模型的三个评估指标:MAE(衡量预测准确性)、MSE(放大大误差)、R²(说明模型解释能力)
算法·机器学习·回归
kokunka3 小时前
【数据结构】栈的顺序存储(整型栈、字符栈)
数据结构
小指纹4 小时前
图论-最短路Dijkstra算法
数据结构·c++·算法·深度优先·图论
赴3355 小时前
逻辑回归 银行贷款资格判断案列优化 交叉验证,调整阈值,下采样与过采样方法
算法·机器学习·逻辑回归·下采样·交叉验证·过采样·阈值