Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class RandomizedSet {
- public:
- /** Initialize your data structure here. */
- RandomizedSet() {
- }
- /** Inserts a value to the set. Returns true if the set did not already contain the specified element. */
- bool insert(int val) {
- if (m.find(val) != m.end()) return false;
- nums.emplace_back(val);
- m[val] = nums.size() - 1;
- return true;
- }
- /** Removes a value from the set. Returns true if the set contained the specified element. */
- bool remove(int val) {
- if (m.find(val) == m.end()) return false;
- int last = nums.back();
- m[last] = m[val];
- nums[m[val]] = last;
- nums.pop_back();
- m.erase(val);
- return true;
- }
- /** Get a random element from the set. */
- int getRandom() {
- return nums[rand() % nums.size()];
- }
- private:
- vector<int> nums;
- unordered_map<int, int> m;
- };
Advertisement
Add Comment
Please, Sign In to add comment