//381. O(1) 时间插入、删除和获取随机元素 - 允许重复
/*
* 使用数组来存储所有元素
* 因为不需要考虑顺序,所以插入的话,直接放在数组后面就可以,常数时间
* 主要是删除,删除的时候,需要先查找元素所在的位置,然后删除
* 查找的时候,可以通过哈希map和set结合,记录各种元素的位置
* 用这种方法可以使查找的时间变为常数时间
* 查找之后,就是删除
* 删除的话可以将当前元素和数组的最后一个元素互换内容,然后删除数组的最后一个元素,时间也是常数
* 删除之后,更新一下元素的位置即可
*/
class RandomizedCollection {
public:
unordered_map<int, unordered_set<int>> id;
vector<int>nums;
/** Initialize your data structure here. */
RandomizedCollection() {
}
/** Inserts a value to the collection. Returns true if the collection did not already contain the specified element. */
bool insert(int val) {
id[val].emplace(nums.size());
nums.emplace_back(val);
return id[val].size() == 1;
}
/** Removes a value from the collection. Returns true if the collection contained the specified element. */
bool remove(int val) {
if (id[val].empty()) return false;
int vid = *(id[val].begin());
nums[vid] = nums.back();
id[val].erase(vid);
id[nums.back()].erase(nums.size() - 1);
if (vid != nums.size() - 1) id[nums.back()].emplace(vid);
nums.pop_back();
return true;
}
/** Get a random element from the collection. */
int getRandom() {
return nums[rand() % nums.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();
*/
LeetCode381. O(1) 时间插入、删除和获取随机元素 - 允许重复
最新推荐文章于 2024-09-22 18:22:06 发布