380. Insert Delete GetRandom O(1)

最新推荐文章于 2022-08-21 22:21:03 发布

zyilove34

最新推荐文章于 2022-08-21 22:21:03 发布

阅读量193

点赞数

分类专栏：笔试题文章标签：数据结构

本文链接：https://blog.youkuaiyun.com/zyilove34/article/details/74019038

版权

笔试题专栏收录该内容

32 篇文章

订阅专栏

本文介绍了一种使用ArrayList实现的随机集合，该集合支持在平均O(1)时间内完成插入、删除及获取随机元素的操作。具体包括：检查元素是否存在并插入、从集合中移除指定元素、从当前集合中返回一个随机元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

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

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

public class RandomizedSet {

/** Initialize your data structure here. */
ArrayList<Integer> array=null;
public RandomizedSet() {
if(array==null){
array=new ArrayList<Integer>();
}

}

/** Inserts a value to the set. Returns true if the set did not already contain the specified element. */
public boolean insert(int val) {
if(array.contains(val)){
return false;
}else{
array.add(val);
return true;
}
}

/** Removes a value from the set. Returns true if the set contained the specified element. */
public boolean remove(int val) {
for(int i=0;i<array.size();i++){
if(array.get(i)==val){
array.remove(i);
return true;
}
}
return false;
}

/** Get a random element from the set. */
public int getRandom() {
if(array==null||array.size()==0){
return -1;
}
int t= (int)(Math.random()*10%array.size());
return array.get(t);
}
}