380. Insert Delete GetRandom O(1)

1

这题难点在于如何在删除的时候做到O1, 因为要随机访问毫无疑问你要用到数组加hash结构,感觉这确实是个新思路

2

新增和随机访问没什么好说的, 来看删除如何做到O1

3

问题 可以在hash里面保存元素对应的索引值, 但是如果删除了数组中间某个元素, 会导致后面的索引全部需要更新, 这样就不是O1了

解决:  假如数组有10个元素 要删除第5个,  不要直接删除第5个, 这样后面的元素索引都会减1,  因此把第五个和最后一个元素交换位置, 同时更新hash里面的两个元素的value;

最后删除数组最后一个元素.

4

如果删除的本来就是最后一个元素, 直接删除就好

class RandomizedSet {
public:
    unordered_map<int,int> kv;
    vector<int> index;
    /** Initialize your data structure here. */
    RandomizedSet() {
        srand(time(NULL));
    }
    
    /** Inserts a value to the set. Returns true if the set did not already contain the specified element. */
    bool insert(int val) {
        if(kv.find(val)!=kv.end())
            return false;
        index.push_back(val);
        kv[val]=index.size()-1;
        return true;
    }
    
    /** Removes a value from the set. Returns true if the set contained the specified element. */
    bool remove(int val) {
        auto iter=kv.find(val);
        if(iter==kv.end())return false;
        if(iter->second<index.size()-1)
        {
            kv[index.back()]=iter->second;
            index[iter->second]=index.back();
        }
            kv.erase(iter);
            index.pop_back();
        return true;
    }
    
    /** Get a random element from the set. */
    int getRandom() {
        return index[rand()%index.size()];
    }
};

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