leetcode 380. O(1) 时间插入、删除和获取随机元素 java

最新推荐文章于 2023-01-31 14:25:31 发布

奔跑的废柴

最新推荐文章于 2023-01-31 14:25:31 发布

阅读量140

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/weixin_42970433/article/details/124156515

版权

LeetCode 专栏收录该内容

279 篇文章 6 订阅

订阅专栏

https://leetcode-cn.com/problems/insert-delete-getrandom-o1/

class RandomizedSet {
    HashMap<Integer,Integer> map;
    ArrayList<Integer> list;
    Random random;
    public RandomizedSet() {
        map=new HashMap<>();
        list=new ArrayList<>();
        random=new Random();
    }
    
    public boolean insert(int val) {
        if(map.containsKey(val)){
            return false;
        }
        else{
            int index=list.size();
            map.put(val,index);
            list.add(val);
            return true;
        }
    }
    
    public boolean remove(int val) {
        if(!map.containsKey(val)){
            return false;
        }
        int index=map.get(val);
        int last=list.get(list.size()-1);
        if(last!=val){
            list.remove(list.size()-1);
            list.set(index,last);
            map.remove(val);
            map.put(last,index);
            return true;
        }
        else{
            list.remove(list.size()-1);
            map.remove(val);
            return true;
        }
    }
    
    public int getRandom() {
        int index=random.nextInt(list.size());
        return list.get(index);
    }
}

/**
 * Your RandomizedSet object will be instantiated and called as such:
 * RandomizedSet obj = new RandomizedSet();
 * boolean param_1 = obj.insert(val);
 * boolean param_2 = obj.remove(val);
 * int param_3 = obj.getRandom();
 */