设计RandomPool结构_random pooling-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_42076938/article/details/107413169

本文介绍了一种名为RandomPool的数据结构，其支持O(1)时间复杂度的插入、删除和随机获取元素操作。通过使用两个映射表，该结构能够确保在不重复插入的情况下，高效地进行元素管理和随机选择。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

设计RandomPool结构

【题目】设计一种结构，在该结构中有如下三个功能：insert(key)：将某个key加入到该结构，做到不重复加入。delete(key)：将原本在结构中的某个key移除。 getRandom()：等概率随机返回结构中的任何一个key。
【要求】 Insert、 delete和getRandom方法的时间复杂度都是O(1)

// 设计RandomPool结构.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include<iostream>
#include<map>
#include<string>
#include<cstdlib>
#include <ctime>
using namespace std;
class RandomPool {
private:
	map<string, int>keyindexMap;
	map<int, string>indexKeyMap;
	int size=0;
public:
	void insert(string key)
	{
		if (!this->keyindexMap.count(key))
		{
			this->keyindexMap[key]=this->size;
			this->indexKeyMap[this->size] = key;
			this->size++;
		}
	}
	void Delete(string key)
	{
		if (keyindexMap.count(key))
		{
			int DeleteIndex = keyindexMap[key];
			int LastIndex = --this->size;
			string LastKey = indexKeyMap[LastIndex];
			indexKeyMap[DeleteIndex] = LastKey;
			keyindexMap[LastKey] = DeleteIndex;
			keyindexMap.erase(key);
			indexKeyMap.erase(LastIndex);
		}
	}
	string getRandow()
	{
		if (this->size == 0) return nullptr;
		int randomIndex = rand()%this->size;
		return indexKeyMap[randomIndex];
	}

};

int main()
{
	RandomPool pool;
	pool.insert("zheng");
	pool.insert("yun");
	pool.insert("tian");
	pool.Delete("yun");
	string a = pool.getRandow();
	string b= pool.getRandow();
	string c= pool.getRandow();
	string d = pool.getRandow();
	cout << a << endl;
	cout << b << endl;
	cout << c<< endl;
	cout << d<< endl;
    return 0;
}