105 lines
2.9 KiB
Java
105 lines
2.9 KiB
Java
//设计一个支持在平均 时间复杂度 O(1) 下，执行以下操作的数据结构。
|
||
//
|
||
//
|
||
// insert(val)：当元素 val 不存在时，向集合中插入该项。
|
||
// remove(val)：元素 val 存在时，从集合中移除该项。
|
||
// getRandom：随机返回现有集合中的一项。每个元素应该有相同的概率被返回。
|
||
//
|
||
//
|
||
// 示例 :
|
||
//
|
||
//
|
||
//// 初始化一个空的集合。
|
||
//RandomizedSet randomSet = new RandomizedSet();
|
||
//
|
||
//// 向集合中插入 1 。返回 true 表示 1 被成功地插入。
|
||
//randomSet.insert(1);
|
||
//
|
||
//// 返回 false ，表示集合中不存在 2 。
|
||
//randomSet.remove(2);
|
||
//
|
||
//// 向集合中插入 2 。返回 true 。集合现在包含 [1,2] 。
|
||
//randomSet.insert(2);
|
||
//
|
||
//// getRandom 应随机返回 1 或 2 。
|
||
//randomSet.getRandom();
|
||
//
|
||
//// 从集合中移除 1 ，返回 true 。集合现在包含 [2] 。
|
||
//randomSet.remove(1);
|
||
//
|
||
//// 2 已在集合中，所以返回 false 。
|
||
//randomSet.insert(2);
|
||
//
|
||
//// 由于 2 是集合中唯一的数字，getRandom 总是返回 2 。
|
||
//randomSet.getRandom();
|
||
//
|
||
// Related Topics 设计 数组 哈希表
|
||
// 👍 314 👎 0
|
||
|
||
package leetcode.editor.cn;
|
||
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
import java.util.Random;
|
||
|
||
//380:常数时间插入、删除和获取随机元素
|
||
public class InsertDeleteGetrandomO1 {
|
||
public static void main(String[] args) {
|
||
//测试代码
|
||
// Solution solution = new InsertDeleteGetrandomO1().new Solution();
|
||
}
|
||
|
||
//力扣代码
|
||
//leetcode submit region begin(Prohibit modification and deletion)
|
||
class RandomizedSet {
|
||
|
||
private List<Integer> list;
|
||
|
||
/**
|
||
* Initialize your data structure here.
|
||
*/
|
||
public RandomizedSet() {
|
||
list = new ArrayList<>();
|
||
}
|
||
|
||
/**
|
||
* Inserts a value to the set. Returns true if the set did not already contain the specified element.
|
||
*/
|
||
public boolean insert(int val) {
|
||
if (list.contains(val)) {
|
||
return false;
|
||
}
|
||
list.add(val);
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* Removes a value from the set. Returns true if the set contained the specified element.
|
||
*/
|
||
public boolean remove(int val) {
|
||
if (!list.contains(val)) {
|
||
return false;
|
||
}
|
||
list.remove((Integer) val);
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* Get a random element from the set.
|
||
*/
|
||
public int getRandom() {
|
||
Random random = new Random();
|
||
return list.get(random.nextInt(list.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();
|
||
*/
|
||
//leetcode submit region end(Prohibit modification and deletion)
|
||
|
||
} |