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

面试刷题第8天

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

1、题目描述

实现RandomizedSet 类：
（1）RandomizedSet() 初始化 RandomizedSet 对象。
（2）bool insert(int val) 当元素 val 不存在时，向集合中插入该项，并返回 true；否则，返回 false 。
（3）bool remove(int val) 当元素 val 存在时，从集合中移除该项，并返回 true；否则，返回 false 。
（4）int getRandom() 随机返回现有集合中的一项（测试用例保证调用此方法时集合中至少存在一个元素）。每个元素应该有相同的概率被返回。
你必须实现类的所有函数，并满足每个函数的平均时间复杂度为O(1) 。

2、解题思路

(1)采用两个HashMap，第一个HashMap是val—>index，第二个HashMap是inedx—>val。size表示HashMap中元素的数量。
（2）插入时，若valIndex中没有此元素，则将此元素插入到两个HashMap中。
（3）删除时，如果简单的执行删除，会导致HashMap中有很多空；采用的方法是：将HashMap中的最后一组元素中的val，替换删除的val，索引还是占用删除的元素的索引，最后一组索引位置删除。
（4）查找时，根据随机生成的索引，在index—>val对应的HashMap中查找。
Java代码如下所示：

class RandomizedSet {
    private HashMap<Integer, Integer> valIndexMap;
    private HashMap<Integer, Integer> indexValMap;
    private int size;
    //初始化
    public RandomizedSet() {
        valIndexMap = new HashMap<Integer, Integer>();
        indexValMap = new HashMap<Integer, Integer>();
        size = 0;
    }
    public boolean insert(int val) {
        if(!valIndexMap.containsKey(val)){
            valIndexMap.put(val, size);
            indexValMap.put(size++, val);
            return true;
        }
        return false;
    }
    public boolean remove(int val) {
        if(valIndexMap.containsKey(val)){
            int deleteIndex = valIndexMap.get(val);
            int lastIndex = --size;
            int lastVal = indexValMap.get(lastIndex);
            valIndexMap.put(lastVal, deleteIndex);
            indexValMap.put(deleteIndex, lastVal);
            valIndexMap.remove(val);
            indexValMap.remove(lastIndex);
            return true;
        }
        return false;
    }
    public int getRandom() {
        if(size == 0){
            return -1;
        }
        int randomIndex = (int)(Math.random()*size);
        return indexValMap.get(randomIndex);
    }
}