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

### 插入、删除和获取随机元素的时间复杂度为 O(1) 的数据结构 为了满足插入、删除以及获取随机元素的操作均能在常数时间内完成，可以采用一种组合型的数据结构来实现这一目标。具体来说，可以通过 **哈希表** 和 **动态数组** 的结合来达到目的。 #### 数据结构设计 - 使用一个 `HashMap` 来存储每个元素及其对应的索引位置。这使得查找操作可以在 O(1) 时间内完成。 - 同时维护一个 `ArrayList` 或类似的动态数组用于支持快速访问任意索引处的元素，并允许在 O(1) 时间内交换最后一个元素与待移除的元素[^3]。 这种设计的核心在于利用哈希表提供高效的键值映射能力，而动态数组则提供了按索引存取的能力。通过这种方式，能够有效地实现在 O(1) 时间内的插入、删除和随机选取功能。 #### 实现方法 以下是基于上述思路的一种可能实现方式： ```java import java.util.*; class RandomizedSet { private Map<Integer, Integer> map; private List<Integer> list; private Random rand; public RandomizedSet() { this.map = new HashMap<>(); this.list = new ArrayList<>(); this.rand = new Random(); } /** Inserts an element into the set */ public boolean insert(int val) { if (map.containsKey(val)) return false; // Already exists. map.put(val, list.size()); // Store index mapping. list.add(val); // Add value to end of arraylist. return true; } /** Removes an element from the set */ public boolean remove(int val) { if (!map.containsKey(val)) return false; int idx = map.get(val); int lastElement = list.get(list.size()-1); // Swap current item with last one in list & update its index in hashmap. Collections.swap(list, idx, list.size()-1); map.put(lastElement, idx); // Remove old references after swapping. list.remove(list.size()-1); map.remove(val); return true; } /** Get random element */ public int getRandom() { return list.get(rand.nextInt(list.size())); } } ``` 此代码片段展示了如何构建这样一个类 (`RandomizedSet`) ，它实现了所需的功能并保持所有关键操作都在恒定时间内执行[^4]。 #### 关于其他潜在解决方案 除了上面提到的方法外，还有几种变体或者扩展方案可供考虑： - 如果不需要处理重复项，则可以直接简化逻辑； - 对于更复杂的场景比如带有权重的概率分布下的随机抽取等问题，则需引入额外机制调整概率计算过程[^5]。