【Leetcode】381. Insert Delete GetRandom O(1) - Duplicates Allowed

题目地址：

https://leetcode.com/problems/insert-delete-getrandom-o1-duplicates-allowed/

要求实现一个数据结构，能在$O(1)$的时间内完成：
1、添加一个数，如果原本存在则返回false，否则返回true。注意，即使已经存在也要添加；
2、删除一个数，如果原本不存在则返回false，否则返回true；
3、随机返回一个数，每个数的返回的概率应该与其出现的频次成正比。

思路参考https://blog.csdn.net/qq_46105170/article/details/109269399。

思路是哈希表 + list。由于要$O(1)$删除，我们需要存储每个数出现的下标。由于这里允许重复元素的存在，所以下标也要存多份。添加操作并不难，在删除的时候，避免错误的方法是，先判断要删除的数是否恰好等于list的最后一个数，如果是，那么很好删除；如果不是，那就确定了要删除的数和list最后一个数不一样，这时候就可以将本题看待成Duplicates not allowed的情形了，做哈希表的修改就不会出错。代码如下：

import java.util.*;

public class RandomizedCollection {
    
    private List<Integer> list;
    private Map<Integer, Set<Integer>> map;
    private Random random;
    
    /**
     * Initialize your data structure here.
     */
    public RandomizedCollection() {
        list = new ArrayList<>();
        map = new HashMap<>();
        random = new Random();
    }
    
    /**
     * Inserts a value to the collection. Returns true if the collection did not already contain the specified element.
     */
    public boolean insert(int val) {
        boolean res = false;
        if (!map.containsKey(val) || map.get(val).isEmpty()) {
            res = true;
        }
        
        list.add(val);
        map.putIfAbsent(val, new HashSet<>());
        map.get(val).add(list.size() - 1);
        
        return res;
    }
    
    /**
     * Removes a value from the collection. Returns true if the collection contained the specified element.
     */
    public boolean remove(int val) {
        if (!map.containsKey(val) || map.get(val).isEmpty()) {
            return false;
        }
    
        Set<Integer> set = map.get(val);
        if (val == list.get(list.size() - 1)) {
            set.remove(list.size() - 1);
            list.remove(list.size() - 1);
        } else {
            int pos = set.iterator().next();
            set.remove(pos);
    
            int last = list.get(list.size() - 1);
            map.get(last).remove(list.size() - 1);
            map.get(last).add(pos);
            Collections.swap(list, pos, list.size() - 1);
            list.remove(list.size() - 1);
        }
        
        return true;
    }
    
    /**
     * Get a random element from the collection.
     */
    public int getRandom() {
        return list.get(random.nextInt(list.size()));
    }
}

所有操作时间复杂度$O(1)$，空间$O(n)$。