当前位置: 首页 > news >正文

力扣380:O(1)时间插入、删除和获取随机数

实现RandomizedSet 类:

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

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

示例:

输入
["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 。

代码:

#define MAX 200000 // 定义最大容量为200000
#define INVALID -1 // 定义无效索引值为-1typedef struct {int val[MAX]; // 存储元素的数组int cnt;      // 当前元素数量
} RandomizedSet;// 创建一个新的RandomizedSet对象
RandomizedSet* randomizedSetCreate() {RandomizedSet *randArray = malloc(sizeof(RandomizedSet)); // 分配内存给RandomizedSet结构体memset(randArray, 0, sizeof(RandomizedSet)); // 将分配的内存初始化为0return randArray; // 返回指向新创建的RandomizedSet对象的指针
}// 向RandomizedSet中插入一个值
bool randomizedSetInsert(RandomizedSet* obj, int val) {int cnt = obj->cnt; // 获取当前元素数量if (cnt == 0) { // 如果当前没有元素obj->val[cnt++] = val; // 将新值插入到数组的第一个位置obj->cnt = cnt; // 更新元素数量} else { // 如果已经有元素for (int i = 0; i < cnt; i++) { // 遍历现有元素if (obj->val[i] == val) { // 如果找到相同的值return false; // 返回false,表示插入失败}}obj->val[cnt++] = val; // 将新值插入到数组的末尾obj->cnt = cnt; // 更新元素数量}return true; // 返回true,表示插入成功
}// 从RandomizedSet中移除一个值
bool randomizedSetRemove(RandomizedSet* obj, int val) {int index = INVALID; // 初始化索引为无效值int cnt = obj->cnt; // 获取当前元素数量if (cnt == 0) { // 如果当前没有元素return false; // 返回false,表示移除失败} else { // 如果已经有元素for (int i = 0; i < cnt; i++) { // 遍历现有元素if (obj->val[i] == val) { // 如果找到要移除的值index = i; // 记录该值的索引break; // 退出循环}}if (index == INVALID) { // 如果未找到要移除的值return false; // 返回false,表示移除失败}for (int i = index; i < cnt - 1; i++) { // 将后续元素前移一位int tmp = obj->val[i + 1]; // 暂存下一个元素的值obj->val[i] = tmp; // 将下一个元素的值赋给当前位置}cnt--; // 减少元素数量obj->cnt = cnt; // 更新元素数量return true; // 返回true,表示移除成功}
}// 从RandomizedSet中随机获取一个值
int randomizedSetGetRandom(RandomizedSet* obj) {int cnt = obj->cnt; // 获取当前元素数量int n = rand() % cnt; // 生成一个0到cnt-1之间的随机数return obj->val[n]; // 返回随机选择的元素
}// 释放RandomizedSet对象的内存
void randomizedSetFree(RandomizedSet* obj) {free(obj); // 释放内存
}

http://www.lryc.cn/news/494520.html

相关文章:

  • 【C++boost::asio网络编程】有关socket的创建和连接的笔记
  • 超级灵感:前端页面功能统一管理方案
  • 力扣第 77 题 组合
  • (超详细图文)PLSQL Developer 配置连接远程 Oracle 服务
  • 元器件选型与参数13 电源的分类-线性电源参数 RT9013 AMS1117 PCB布局布线
  • RHEL7+Oracle11.2 RAC集群-多路径(multipath+udev)安装步骤
  • 每日速记10道java面试题03
  • Vue 3 的双向绑定原理
  • 如何使用 Chrome 无痕浏览模式访问网站?
  • Idea 2024.3 突然出现点击run 运行没有反应,且没有任何提示。
  • 【小白学机器学习36】关于独立概率,联合概率,交叉概率,交叉概率和,总概率等 概念辨析的例子
  • Spring Boot 项目——分层架构
  • wordpress网站首页底部栏显示网站备案信息
  • python面向对象编程练习
  • OpenCV_Code_LOG
  • 力扣第 74 题是 搜索二维矩阵
  • [极客大挑战 2019]BabySQL--详细解析
  • 实现Linux平台自定义协议族
  • RL78/G15 Fast Prototyping Board Arduino IDE 平台开发过程
  • YOLOv11 NCNN安卓部署
  • 对载入的3dtiles进行旋转、平移和缩放变换。
  • Rust个人认为将抢占C和C++市场,逐渐成为主流的开发语言
  • 在openEuler中使用top命令
  • 探索文件系统,Python os库是你的瑞士军刀
  • 【小白学机器学习41】如何从正态分布的总体中去抽样? 获得指定正态分布的样本的2种方法
  • 将VSCode设置成中文语言环境
  • Applied Intelligence投稿
  • AI-agent矩阵营销:让品牌传播无处不在
  • 【0346】Postgres内核 Startup Process 通过 signal 与 postmaster 交互实现 (5)
  • NSSCTF-做题笔记