LeetCode O(1) 时间插入、删除和获取随机元素
变长数组 + 哈希表
这道题要求实现一个类,满足插入、删除和获取随机元素操作的平均时间复杂度为 O(1)。
变长数组可以在 O(1) 的时间内完成获取随机元素操作,但是由于无法在 O(1) 的时间内判断元素是否存在,因此不能在 O(1) 的时间内完成插入和删除操作。哈希表可以在 O(1) 的时间内完成插入和删除操作,但是由于无法根据下标定位到特定元素,因此不能在 O(1) 的时间内完成获取随机元素操作。为了满足插入、删除和获取随机元素操作的时间复杂度都是 O(1),需要将变长数组和哈希表结合,变长数组中存储元素,哈希表中存储每个元素在变长数组中的下标。
插入操作
插入操作时,首先判断 val
是否在哈希表中,如果已经存在则返回 false
,如果不存在则插入 val
,操作如下:
- 在变长数组的末尾添加
val
。 - 在添加
val
之前的变长数组长度为val
所在下标index
,将val
和下标index
存入哈希表。 - 返回
true
。
删除操作
删除操作时,首先判断 val
是否在哈希表中,如果不存在则返回 false
,如果存在则删除 val
,操作如下:
- 从哈希表中获得
val
的下标index
。 - 将变长数组的最后一个元素
last
移动到下标index
处,在哈希表中将last
的下标更新为index
。 - 在变长数组中删除最后一个元素,在哈希表中删除
val
。 - 返回
true
。
删除操作的重点在于将变长数组的最后一个元素移动到待删除元素的下标处,然后删除变长数组的最后一个元素。该操作的时间复杂度是 O(1),且可以保证在删除操作之后变长数组中的所有元素的下标都连续,方便插入操作和获取随机元素操作。
获取随机元素操作
由于变长数组中的所有元素的下标都连续,因此随机选取一个下标,返回变长数组中该下标处的元素。
class RandomizedSet {
List<Integer> nums;
Map<Integer, Integer> indices;
Random random;
public RandomizedSet() {
nums = new ArrayList<Integer>();
indices = new HashMap<Integer, Integer>();
random = new Random();
}
public boolean insert(int val) {
if (indices.containsKey(val)) {
return false;
}
int index = nums.size();
nums.add(val);
indices.put(val, index);
return true;
}
public boolean remove(int val) {
if (!indices.containsKey(val)) {
return false;
}
int index = indices.get(val);
int last = nums.get(nums.size() - 1);
nums.set(index, last);
indices.put(last, index);
nums.remove(nums.size() - 1);
indices.remove(val);
return true;
}
public int getRandom() {
int randomIndex = random.nextInt(nums.size());
return nums.get(randomIndex);
}
}
1 评论
en0sw9