1057-插入、删除和随机访问都是 O(1) 的容器

题目如下

设计一个支持在平均 时间复杂度 O(1) 下，执行以下操作的数据结构：

insert(val)：当元素 val 不存在时返回 true ，并向集合中插入该项，否则返回 false 。
remove(val)：当元素 val 存在时返回 true ，并从集合中移除该项，否则返回 false 。
getRandom：随机返回现有集合中的一项。每个元素应该有 相同的概率 被返回。

解题思路

思路——————哈希表 + 数组
由于题目要求删除和插入的时间复杂度为 O(1)，能够满足要求的只有哈希表，显然此处需要使用哈希表。

但是使用哈希表无法等概率返回其中的每个数值。若这些数值是保存在数组内，则可以轻松解决。但若只使用数组，那么删除和插入一个值时，都需遍历一遍数组找到数值对应的下标，则会使时间复杂度为 O(n)。

故可将哈希表和数组结合，数组内存数值实现 getRandom，哈希内存数值和下标的映射实现 insert 和 remove 的索引查找的时间复杂度为 O(1)，之后在数组内操作数值的插入和删除。具体来看操作就是：

1. 插入
每次添加新数值时，先使用哈希表判断该数值是否存在，存在则直接返回false。不存在则进行插入操作，只要将该数值添加到数组尾部即可，并将该数值和其下标的映射存入哈希表。

2. 删除
删除同样需使用哈希表判断是否存在，若不存在则返回false。存在则进行删除操作，在哈希表中删除时间复杂度为 O(1)，但是在数值中删除比较麻烦。若只是直接删除，则为了保证数组内存连续性需将删除数值后面的数值均前移一位，时间复杂度为 O(n)。比较好的处理方式是，用数组的最后一个数值去填充需要删除的数值的内存，其他数值在数组中的位置保持不变，并将这个拿来填充的数值的下标更新即可，最后只要删除数组最后一个数值，同样可以保证时间复杂度为 O(1)。

3. 随机返回
只要随机生成数组下标范围内一个随机下标值，返回该数组下标内的数值即可。

解题代码

class RandomizedSet 
{
private:
    unordered_map<int, int> numToLocation;
    vector<int> nums;
public:
    RandomizedSet() 
    {

    }

    bool insert(int val) 
    {
        if (numToLocation.count(val)) 
        {
            return false;
        }
        numToLocation[val] = nums.size();
        nums.push_back(val);
        return true;
    }

    bool remove(int val) 
    {
        if (!numToLocation.count(val)) 
        {
            return false;
        }
        int location = numToLocation[val];
        numToLocation[nums.back()] = location;
        numToLocation.erase(val);
        nums[location] = nums.back();
        nums.pop_back();
        return true;
    }

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