设计一个支持全O(1)的插入、删除和随机获取的集合
集合(Set)作为存储数据的容器时,不允许存储相同的元素,只能保留一份,能够快速帮助我们进行去重操作,过滤重复的元素。
插入(Insert):在 O(1) 时间内将一个元素插入集合中。
删除(Remove):在 O(1) 时间内从集合中删除一个元素。
随机获取(GetRandom):在 O(1) 时间内从集合中随机获取一个元素。
对于插入和删除操作,常见的数据结构是哈希表,因为哈希表可以在平均情况下提供 O(1) 的插入和删除操作。但是,对于随机获取操作,哈希表并不能直接提供 O(1) 的性能,因为哈希表中元素的位置是由哈希函数决定的,无法直接进行随机访问。
因此,我们需要一个数据结构,既能提供哈希表的快速插入和删除,又能够支持随机获取。一种常见的做法是使用数组,因为数组可以实现 O(1) 时间的随机访问。但是,直接使用数组的话,删除操作的平均时间复杂度就不再是 O(1)。
解决这个问题的一种方法是使用两个数据结构的结合:哈希表 + 数组。具体地,使用一个哈希表来存储元素到其在数组中的索引的映射,同时使用一个动态数组来存储元素。这样,插入和删除的操作可以在哈希表中完成,而随机获取的操作可以在数组中完成。
在具体实现中,插入操作需要在数组中添加元素,同时更新哈希表中的映射;删除操作需要在数组中删除元素,同时更新哈希表中的映射;随机获取操作可以通过在数组中随机选择一个索引,然后返回对应的元素来实现。
这就是这个问题的一种解法,其中哈希表和数组的结合使得插入、删除和随机获取这三种操作的平均时间复杂度都是 O(1)。
using System; using System.Collections.Generic; class RandomizedSet { private List<int> nums; private Dictionary<int, int> numToIndex; public RandomizedSet() { nums = new List<int>(); numToIndex = new Dictionary<int, int>(); } public bool Insert(int val) { if (numToIndex.ContainsKey(val)) { return false; // 已存在 } nums.Add(val); numToIndex[val] = nums.Count - 1; return true; } public bool Remove(int val) { if (!numToIndex.ContainsKey(val)) { return false; // 不存在 } int lastNum = nums[nums.Count - 1]; int index = numToIndex[val]; // 将要删除的元素与最后一个元素交换 nums[index] = lastNum; numToIndex[lastNum] = index; // 删除最后一个元素 nums.RemoveAt(nums.Count - 1); numToIndex.Remove(val); return true; } public int GetRandom() { Random rand = new Random(); int randomIndex = rand.Next(nums.Count); return nums[randomIndex]; } } class Program { static void Main() { RandomizedSet set = new RandomizedSet(); Console.WriteLine(set.Insert(1)); // 返回 true Console.WriteLine(set.Remove(2)); // 返回 false Console.WriteLine(set.Insert(2)); // 返回 true Console.WriteLine(set.GetRandom()); // 随机返回 1 或 2 Console.WriteLine(set.Remove(1)); // 返回 true Console.WriteLine(set.Insert(2)); // 返回 false,已存在 Console.WriteLine(set.GetRandom()); // 随机返回 2 } }
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 单线程的Redis速度为什么快?
· SQL Server 2025 AI相关能力初探
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 展开说说关于C#中ORM框架的用法!