【Leetcode】380. Insert Delete GetRandom O(1)

380. Insert Delete GetRandom O(1)

Design a data structure that supports all following operations in average O(1) time.

1. insert(val): Inserts an item val to the set if not already present.
2. remove(val): Removes an item val from the set if present.
3. getRandom: Returns a random element from current set of elements. Each element must have the same probability of being returned.

Example:

// Init an empty set.
RandomizedSet randomSet = new RandomizedSet();

// Inserts 1 to the set. Returns true as 1 was inserted successfully.
randomSet.insert(1);

// Returns false as 2 does not exist in the set.
randomSet.remove(2);

// Inserts 2 to the set, returns true. Set now contains [1,2].
randomSet.insert(2);

// getRandom should return either 1 or 2 randomly.
randomSet.getRandom();

// Removes 1 from the set, returns true. Set now contains [2].
randomSet.remove(1);

// 2 was already in the set, so return false.
randomSet.insert(2);

// Since 2 is the only number in the set, getRandom always return 2.
randomSet.getRandom();

Subscribe to see which companies asked this question.

题目简述：

对输入的数据实现，随机读取，插入，删除。。。并且复杂度是O(1)

思路简述：

看了网上的一些blog,大致的思路是 数组+哈希表。

把输入的数据存成两份，一份数组，一份哈希表。在哈希表上存的时候，key=输入的元素,value=元素在数组的索引.这样，删除的时候先去哈希表上找到元素在数组中的索引，然后再去数组中删除(ps:删除的时候，让数组最后一个元素与该元素对调然后删除最后一个元素)，就实现了删除操作O(1)。同样，插入O(1)。利用数组特性，随机读取O(1)；

代码：

import java.util.Vector;
public class RandomizedSet {
    Map<Integer,Integer> aMap;
    Vector<Integer> aVec;
    //int index;
    
    public RandomizedSet() {
        aMap= new HashMap<Integer,Integer>();
        aVec = new Vector<Integer>();
        //index = 0;
    }
    
    
    public boolean insert(int val) {
        int temp=aVec.size();
        if(!aMap.containsKey(val)){
            aMap.put(val,aVec.size());
            aVec.add(val);
            if(aVec.size()!=temp+1){
                System.out.println("error");
            }
            return true;
        }
        return false;
    }
    
   
    public boolean remove(int val) {
        if(aMap.containsKey(val)){
            int temp = aMap.get(val);
            aVec.set(temp,aVec.get(aVec.size()-1));
            aMap.put(aVec.get(temp),temp);
            
            aMap.remove(val);
            aVec.remove(aVec.size()-1);
            return true;
        }
        return false;
    }
    
    
    public int getRandom() {
        Random rand = new Random();
        //int num = ;
        return aVec.get(rand.nextInt(aVec.size()));
    }
}

运行结果：