设计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);
}