【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、随机返回一个数，每个数的返回的概率应该与其出现的频次成正比。

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

class RandomizedCollection {
public:
  vector<int> v;
  unordered_map<int, unordered_set<int>> mp;

  RandomizedCollection() {}

  bool insert(int x) {
    auto [it, ins] = mp.try_emplace(x);
    v.push_back(x);
    it->second.insert(v.size() - 1);
    return ins;
  }

  bool remove(int x) {
    auto it = mp.find(x);
    if (it == mp.end()) return false;
    auto &pos_set = it->second;
    if (x == v.back()) {
      pos_set.erase(v.size() - 1);
      v.pop_back();
      if (pos_set.empty()) mp.erase(it);
    } else {
      int pos = *pos_set.begin();
      pos_set.erase(pos_set.begin());
      if (pos_set.empty()) mp.erase(it);
      int last = v.back();
      auto it_last = mp.find(last);
      it_last->second.erase(v.size() - 1);
      it_last->second.insert(pos);
      v[pos] = last;
      v.pop_back();
    }
    return true;
  }

  int getRandom() {
    return v[rand() % v.size()];
  }
};

/**
 * Your RandomizedCollection object will be instantiated and called as such:
 * RandomizedCollection* obj = new RandomizedCollection();
 * bool param_1 = obj->insert(val);
 * bool param_2 = obj->remove(val);
 * int param_3 = obj->getRandom();
 */

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