设计RandomPool结构

设计RandomPool结构

题目：

设计一种结构，在该结构中有如下三种功能，insert(key)：将key加入到该结构，做到不重复加入。delete(key)：将key删除。getRandom()：等概率随机返回结构中的任何一个key。
要求：三种方法的时间复杂度为O(1)。

思路：

设计两个hash表，第一个hash表对应(key, value)，第二个哈希表对应(value, key)。再设置一个变量size，表示hash表中有多少元素。

private HashMap<K, Integer> keyIndexMap;
private HashMap<Integer, K> indexKeyMap;
private int size;

//初始化
public Pool()
{
	this.keyIndexMap = new HashMap<K, Integer>();
	this.indexKeyMap = new HashMap<Integer, K>();
	this.size = 0;
}

insert操作

public void insert(K key)
{
	if(!this.keyIndexMap.containsKey(key))
	{
		this.keyIndexMap.put(key, size);
		this.indexKeyMap.put(size, key);
		size++;
	}
}

delete操作

删除操作中，为了保证以O(1)随机返回任何一个key，这里在删除任何一个key时，都用最后一个key代替这个key，然后删除最后一个key。

public void delete(K key)
{
	if(this.keyIndexMap.containsKey(key))
	{
		int deleteIndex = this.keyIndexMap.get(key);//删除位置的索引
		int lastIndex = --this,size;//最后一个key的索引
		K lastKey = this.indexKeyMap.get(lastIndex);//最后一个key
		this.keyIndexMap.put(lastKey, deleteIndex);
		this.indexKeyMap.put(deleteIndex, lastKey);
		this.keyIndexMap.remove(key);
		this.indexKeyMap.remove(lastIndex);
	}
}

getRandom操作

public K getRandom()
{
	if(this.size == 0)
	{
		return null;
	}
	int randomIndex = (int)(Math.random() * this.size);//0-size-1
	return this.indexKeyMap.get(randomIndex);
}