算法记录
LeetCode 题目:
实现RandomizedSet 类
说明
一、题目
你必须实现类的所有函数,并满足每个函数的 平均 时间复杂度为 O(1) 。
二、分析
- 每个函数的平均时间复杂时间为常数,也就是要使用
.map
来完成数据的存储。 - 但是映射集合又不支持随机访问遍历,这里我们就可以使用一个数组去完成输出的随机输出。
- 只需要在更新数据时同时更新数组和映射集合即可。
class RandomizedSet {
private Map<Integer, Integer> map;
private List<Integer> flag;
private Random random;
public RandomizedSet() {
map = new HashMap();
flag = new ArrayList();
random = new Random();
}
public boolean insert(int val) {
if(map.get(val) != null) return false;
map.put(val, flag.size());
flag.add(val);
return true;
}
public boolean remove(int val) {
if(map.get(val) == null) return false;
int index = map.get(val);
int ret = flag.get(flag.size() - 1);
flag.set(index, ret);
flag.remove(flag.size() - 1);
map.remove(val);
if(map.get(ret) != null) map.put(ret, index);
return true;
}
public int getRandom() {
int ran = random.nextInt(flag.size());
return flag.get(ran);
}
}
总结
熟悉用 hashMap
提高数组的加速查询。