设计RandomPool结构

题目描述

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

思路

维护两个hashmap,一个存放<key,Integer>，一个存放 <key,Integer>
size记录当前有多少个key
插入时在两个map中同时分别插入<key,size>,<size,key>，然后size加1
删除key时，用最后一个记录填充要删除的key所在的位置，size减1
随机返回时，生成一个0到size范围内的随机数index，返回index位置的key

代码

package LinearStructure;

import java.util.HashMap;

public class RandomPool {

    public static class Pool<K>{
        private HashMap<K,Integer> keyIndexMap;
        private HashMap<Integer,K> indexKeyMap;
        private int size;

        public Pool(){
            this.keyIndexMap=new HashMap<>();
            this.indexKeyMap=new HashMap<>();
            this.size=0;
        }

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

        public void delete(K key){
            if (this.keyIndexMap.containsKey(key)){
                int deleteIndex=this.keyIndexMap.get(key);
                int lastIndex=--this.size;
                K lastKey=this.indexKeyMap.get(lastIndex);
                this.keyIndexMap.put(lastKey, deleteIndex);
                this.indexKeyMap.put(deleteIndex, lastKey);
                this.keyIndexMap.remove(lastKey);
                this.indexKeyMap.remove(lastIndex);
            }
        }

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

    public static void main(String[] args) {
        Pool<String> pool = new Pool<String>();
        pool.insert("zuo");
        pool.insert("cheng");
        pool.insert("yun");
        System.out.println(pool.getRandom());
        System.out.println(pool.getRandom());
        System.out.println(pool.getRandom());
        System.out.println(pool.getRandom());
        System.out.println(pool.getRandom());
        System.out.println(pool.getRandom());
    }

}