O(1) 时间插入、删除和获取随机元素（C++解法）

题目

实现 RandomizedSet 类：

• RandomizedSet() 初始化 RandomizedSet 对象
• bool insert(int val) 当元素 val 不存在时，向集合中插入该项，并返回 true ；否则，返回 false 。
• bool remove(int val) 当元素 val 存在时，从集合中移除该项，并返回 true ；否则，返回 false 。
• int getRandom() 随机返回现有集合中的一项（测试用例保证调用此方法时集合中至少存在一个元素）。每个元素应该有 相同的概率 被返回。

你必须实现类的所有函数，并满足每个函数的 平均 时间复杂度为 O(1) 。

示例：

输入
["RandomizedSet", "insert", "remove", "insert", "getRandom", "remove", "insert", "getRandom"]
[[], [1], [2], [2], [], [1], [2], []]
输出
[null, true, false, true, 2, true, false, 2]

解释
RandomizedSet randomizedSet = new RandomizedSet();
randomizedSet.insert(1); // 向集合中插入 1 。返回 true 表示 1 被成功地插入。
randomizedSet.remove(2); // 返回 false ，表示集合中不存在 2 。
randomizedSet.insert(2); // 向集合中插入 2 。返回 true 。集合现在包含 [1,2] 。
randomizedSet.getRandom(); // getRandom 应随机返回 1 或 2 。
randomizedSet.remove(1); // 从集合中移除 1 ，返回 true 。集合现在包含 [2] 。
randomizedSet.insert(2); // 2 已在集合中，所以返回 false 。
randomizedSet.getRandom(); // 由于 2 是集合中唯一的数字，getRandom 总是返回 2 。

C++代码

#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;

/*
* 定义一个函数可以插入、删除和获取随机元素
* 定义数组和无序map容器
* 用srand()函数设置rand()函数的种子
* 定义插入函数，判断元素是否在哈希表中，将元素插入末尾，将val,index键值对存入哈希表
* 定义删除函数，判断元素是否在哈希表中，获取index值，将last元素移动到下标index处，
更新哈希表中last的下标，删除最后一个元素和哈希表中val键值对
* 获取随机元素
*/
class RandomizedSet {
public:
	RandomizedSet() {
		srand((unsigned)time(NULL));
	}
	bool insert(int val) {
		if (indices.count(val)) {
			return false;
		}
		int index = nums.size();
		nums.emplace_back(val);
		indices[val] = index;
		return true;
	}
	bool remove(int val) {
		if (!indices.count(val)) {
			return false;
		}
		int index = indices[val];
		int last = nums.back();
		nums[index] = last;
		indices[last] = index;
		nums.pop_back();
		indices.erase(val);
		return true;
	}

	int getRandom() {
		int randomIndex = rand() % nums.size();
		return nums[randomIndex];
	}

private:
	vector<int> nums;
	unordered_map<int, int> indices;
};

int main() {
	RandomizedSet* random = new RandomizedSet();
	bool ans1 = random->insert(1);
	bool ans2 = random->remove(2);
	bool ans3 = random->insert(2);
	int ans4 = random->getRandom();
	bool ans5 = random->remove(1);
	bool ans6 = random->insert(2);
	int ans7 = random->getRandom();
	cout << boolalpha << ans1 << " " << boolalpha << ans2 << " " 
		<< boolalpha << ans3 << " " << boolalpha << ans4 << " " 
		<< boolalpha << ans5 << " " << boolalpha << ans6 << " " << boolalpha << ans7 << endl;
	return 0;
}

分析

定义一个类 RandomizedSet 可以插入、删除和获取随机元素，定义数组 nums 和无序 map 容器 indices，用 srand() 函数设置 rand() 函数的种子。定义插入函数 insert()，判断元素是否在哈希表中，将元素插入末尾，将 val , index 键值对存入哈希表；定义删除函数 remove()，判断元素是否在哈希表中，获取 index 值，将 last 元素移动到下标 index 处，更新哈希表中 last 的下标，删除最后一个元素和哈希表中 val 键值对。定义获取随机元素函数。

问题

生成随机元素 rand() 函数和设置 rand() 函数的种子的 srand() 函数。通常使用 time() 函数作为 srand() 函数的输入参数。time() 函数会返回1970年1月1日至今所经历的时间（以秒为单位）。在使用 rand() 函数之前，srand() 函数要先被调用，并且在整个程序中只需被调用一次。

unordered_map() 函数的用法：http://c.biancheng.net/view/7231.html

其中我们使用的 count() 方法可以在容器中查找以 key 键的键值对的个数，erase() 方法可以删除指定键值对。

对于数组的操作，emplace_back() 方法可以在末端插入元素，pop_back() 方法可以删除最后一个元素，back() 方法获取数组最后一个元素。