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

算法记录

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 提高数组的加速查询。