LeetCode 剑指 Offer II 030. 插入、删除和随机访问都是 O(1) 的容器
文章目录
题目描述
设计一个支持在平均 时间复杂度 O(1) 下,执行以下操作的数据结构:
insert(val):当元素 val 不存在时返回 true ,并向集合中插入该项,否则返回 false 。
remove(val):当元素 val 存在时返回 true ,并从集合中移除该项,否则返回 false 。
getRandom:随机返回现有集合中的一项。每个元素应该有 相同的概率 被返回。
示例 :
输入: inputs = [“RandomizedSet”, “insert”, “remove”, “insert”, “getRandom”, “remove”, “insert”, “getRandom”]
[[], [1], [2], [2], [], [1], [2], []]
输出: [null, true, false, true, 2, true, false, 2]
剑指 Offer II 030. 插入、删除和随机访问都是 O(1) 的容器
提示:
-231 <= val <= 231 - 1
最多进行 2 * 105 次 insert , remove 和 getRandom 方法调用
当调用 getRandom 方法时,集合中至少有一个元素
一、解题关键词
二、解题报告
1.思路分析
2.时间复杂度
3.代码示例
class RandomizedSet {
List<Integer> nums;
Map<Integer,Integer> indexs;
Random random;
/** Initialize your data structure here. */
public RandomizedSet() {
nums = new ArrayList<Integer>();
indexs = new HashMap<>();
random = new Random();
}
/** Inserts a value to the set. Returns true if the set did not already contain the specified element. */
public boolean insert(int val) {
if(indexs.containsKey(val)){
return false;
}
int index = nums.size();
nums.add(val);
indexs.put(val,index);
return true;
}
public boolean remove(int val) {
if(!indexs.containsKey(val)){
return false;
}
int index = indexs.get(val);
int last = nums.get(nums.size() - 1);
nums.set(index,last);
indexs.put(last,index);
nums.remove(nums.size() - 1);
indexs.remove(val);
return true;
}
public int getRandom() {
return nums.get(random.nextInt(nums.size()));
}
}
2.知识点