设计RandomPool结构

    设计RandomPool结构
    【题目】 设计一种结构，在该结构中有如下三个功能：
    insert(key)：将某个key加入到该结构，做到不重复加入。
    delete(key)：将原本在结构中的某个key移除。 getRandom()：
    等概率随机返回结构中的任何一个key。

    【要求】 Insert、delete和getRandom方法的时间复杂度都是O(1)import java.util.HashMap;
import java.util.Random;
public class C01_RandomPoll {
public static class Pool<Key>{
//HashMap<K, String>这里的K表示只要是同一种类型就OK了，但并没有指定是哪一种类型
//例如我需要放String的类型，那第一个放进去的时候，K就被指定为String 了
//理论上是更为强大的
HashMap<Key, Integer>keyIndexMap;
HashMap<Integer, Key>indexKeyMap;
int size;
public Pool(){
keyIndexMap = new HashMap<Key, Integer>();
indexKeyMap = new HashMap<Integer, Key>();
}
//插入到pool
public void insert(Key k){
keyIndexMap.put(k, size);
indexKeyMap.put(size++, k);
}
//删除（这个结合文末的图片提示来看）
public void delete(Key key){
if(!keyIndexMap.containsKey(key)){
return ;
}
int lastIndex = --this.size;
int deleteIndex = keyIndexMap.get(key);
Key lastKey = indexKeyMap.get(lastIndex);
keyIndexMap.put(lastKey, deleteIndex);
indexKeyMap.put(deleteIndex, lastKey);
keyIndexMap.remove(lastKey);
}
//获取随机
public Key getRandom(){
if(size==0){
return null;
}
Random random = new Random();
int index = random.nextInt(keyIndexMap.size());
return indexKeyMap.get(index);
}
//test
public static void main(String[] args) {
Pool<String> pool = new Pool<String>();
pool.insert("laoganma");
pool.insert("nengandie");
pool.insert("yyziii");
pool.insert("aaaaaa");
for (int i = 0; i < 10; i++) {
System.out.println(pool.getRandom());
}
pool.delete("nengandie");
System.out.println("=======");
for (int i = 0; i < 10; i++) {
System.out.println(pool.getRandom());
}
}
}
}