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

class RandomizedSet {

    private Map<Integer,Integer> map;
    private List<Integer> list;

    public RandomizedSet() {
        map = new HashMap<>();
        list = new ArrayList<>();
    }

    public boolean insert(int val) {
        if(map.containsKey(val)){
            return false;
        }
        list.add(val);
        map.put(val,list.size()-1);
        return true;
    }

    public boolean remove(int val) {
        if(!map.containsKey(val)){
            return false;
        }
        //拿需要删除元素对应下标
        Integer index = map.get(val);
        //获取list最后一个元素值
        Integer lastValue = list.get(list.size() - 1);
        //把删除元素对应的下标值修改为list最后一个值
        list.set(index,lastValue);
        //删除最后一个元素，O(1)时间复杂度
        list.remove(list.size()-1);
        //设置最后一个值的下标
        map.put(lastValue,index);
        //从map删除要删除的元素
        map.remove(val);
        return true;
    }

    public int getRandom() {
        return list.get(new Random().nextInt(list.size()));
    }
}

代码优化，list换array，array大小固定2*10的5次方


class RandomizedSet {

    private Map<Integer, Integer> map;
    private int[] arr;
    private int index;
    private Random random;

    public RandomizedSet() {
        map = new HashMap<>();
        arr = new int[2 * (int) Math.pow(10, 5)];
        index = 0;
        random = new Random();
    }

    public boolean insert(int val) {
        if (map.containsKey(val)) {
            return false;
        }
        arr[index++] = val;
        map.put(val, index - 1);
        return true;
    }

    public boolean remove(int val) {
        if (!map.containsKey(val)) {
            return false;
        }
        //拿需要删除元素对应下标
        Integer removeIndex = map.get(val);
        //获取list最后一个元素值
        Integer lastValue = arr[index - 1];
        //把删除元素对应的下标值修改为list最后一个值
        arr[removeIndex] = lastValue;
        //删除最后一个元素，O(1)时间复杂度
        index--;
        //设置最后一个值的下标
        map.put(lastValue, removeIndex);
        //从map删除要删除的元素
        map.remove(val);
        return true;
    }

    public int getRandom() {
        return arr[random.nextInt(index)];
    }
}