LeetCode 380. Insert Delete GetRandom O(1)

最新推荐文章于 2024-09-18 11:04:20 发布

Cloud陈

最新推荐文章于 2024-09-18 11:04:20 发布

阅读量233

点赞数

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

本文链接：https://blog.csdn.net/shengyingpo/article/details/58621358

版权

LeetCode 专栏收录该内容

161 篇文章 0 订阅

订阅专栏

public class RandomizedSet {
    Random random;
	Set<Integer> set;

    /** Initialize your data structure here. */
    public RandomizedSet() {
        random = new Random();
        set = new HashSet<Integer>();
    }
    
    /** Inserts a value to the set. Returns true if the set did not already contain the specified element. */
    public boolean insert(int val) {
        return set.add(val);
    }
    
    /** Removes a value from the set. Returns true if the set contained the specified element. */
    public boolean remove(int val) {
        return set.remove(val);
    }
    
    /** Get a random element from the set. */
    public int getRandom() {
        return (Integer)set.toArray()[random.nextInt(set.size())];
    }
}

/**
 * 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();
 */