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

HashSet和单独用ArrayList均不能满足时间需求

哈希表+列表

• 哈希表可以实现O(1) 时间插入、删除
• 列表可以实现O(1) 时间获取随机元素
• 哈希表的value记录的是元素在list中的下标

public class RandomizedSet {
    private List<Integer> list;
    private HashMap<Integer, Integer> map;
    private Random random;

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

    public boolean insert(int val) {
       if (map.containsKey(val)) {
           return false;
       }
       map.put(val, list.size()); // value为val在list中的索引
       list.add(val);
       return true;
    }

    public boolean remove(int val) {
       if(!map.containsKey(val)) {
           return false;
       }
       int index = map.get(val);
       map.remove(val);

        int a=list.removeLast(); // 移除最后一个元素
        if(a != val){ // 移除的不是目标元素
             map.put(a, index); // 把a放到index的位置
             list.set(index, a); //把a放到list中index的位置：实现了类似于先将目标数和最后一个数交换位置然后再删除最后一个数的效果
        }

        return true;
    }

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