当前位置 博文首页 > Michael是个半路程序员：LeetCode 380. 常数时间插入、删除和获
设计一个支持在平均 时间复杂度 O(1) 下，执行以下操作的数据结构。
insert(val)：当元素 val 不存在时，向集合中插入该项。
remove(val)：元素 val 存在时，从集合中移除该项。
getRandom：随机返回现有集合中的一项。每个元素应该有相同的概率被返回。
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/insert-delete-getrandom-o1
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
class RandomizedSet {
vector<int> v;
unordered_map<int,int> m;
public:
RandomizedSet() { }
bool insert(int val) {
if(m.find(val) == m.end())
{
m[val] = v.size();
v.push_back(val);
return true;
}
else
return false;
}
bool remove(int val) {
if(m.find(val) == m.end())
return false;
v[m[val]] = v.back();
m[v.back()] = m[val];
v.pop_back();
m.erase(val);
return true;
}
int getRandom() {
if(v.size() == 0)
return -1;
return v[rand() % v.size()];
}
};
