Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
insert-delete-getrandom-o1-duplicates-allowed.cpp 1.43 KB
一键复制 编辑 原始数据 按行查看 历史
// Time: O(1)
// Space: O(n)
class RandomizedCollection {
public:
/** 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) {
bool has = used_.count(val);
list_.emplace_back(val, used_[val].size());
used_[val].emplace_back(list_.size() - 1);
return !has;
}
/** Removes a value from the collection. Returns true if the collection contained the specified element. */
bool remove(int val) {
if (!used_.count(val)) {
return false;
}
used_[list_.back().first][list_.back().second] = used_[val].back();
swap(list_[used_[val].back()], list_.back());
used_[val].pop_back();
if (used_[val].empty()) {
used_.erase(val);
}
list_.pop_back();
return true;
}
/** Get a random element from the collection. */
int getRandom() {
return list_[rand() % list_.size()].first;
}
private:
vector<pair<int, int>> list_;
unordered_map<int, vector<int>> used_;
};
/**
* 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();
*/
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助