Leetcode 380. Insert Delete GetRandom O(1)

Using ArrayList's get function to implement getRandom() in constant time, and a hashmap to save the <Value, Index> pair.

So we need to make sure the index of the last element always equals to the size of the list - 1.

When we delete an element from the the list, we don't directly delete it, however, replace it with the last element in the list, and update the index in the hash map for this new added element. 

If the insert operation allow duplicate element, add a HashSet to record the indices for duplicate elements. The map would be <Value, HashSet>.

public class RandomizedSet {
    HashMap<Integer, Integer> map;      // A hashtable to save <Value, Index> pair so that allows getRandom() in constant time
    List<Integer> list;                 // A list to allow insert and remove operations
    Random rand;
    
    /** Initialize your data structure here. */
    public RandomizedSet() {
        map = new HashMap<Integer, Integer>();
        list = new ArrayList<Integer>();       // size, get, set all in constant time for ArrayList
        rand = new Random();
    }
    
    /** Inserts a value to the set. Returns true if the set did not already contain the specified element. */
    public boolean insert(int val) {
        if (map.containsKey(val)) {
            return false;
        }
        list.add(val);
        map.put(val, list.size()-1);
        return true;
    }
    
    /** Removes a value from the set. Returns true if the set contained the specified element. */
    public boolean remove(int val) {
        // Remove element from both map and list
        if (!map.containsKey(val)) {
            return false;
        }
        int index = map.get(val);
        int last = list.get(list.size()-1);
        list.set(index, last);                      // replace the element with last element in the list
        list.remove(list.size()-1);                 // remove the element from list
        map.put(last, index);                       // update the index for the last element
        map.remove(val);                            // remove the element from map
        return true;
    }
    
    /** Get a random element from the set. */
    public int getRandom() {
        return list.get(rand.nextInt(list.size()));     // return a random position from 0 to list.size()-1
    }
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值