题目地址:
https://leetcode.com/problems/insert-delete-getrandom-o1-duplicates-allowed/
要求实现一个数据结构,能在
O
(
1
)
O(1)
O(1)的时间内完成:
1、添加一个数,如果原本存在则返回false,否则返回true。注意,即使已经存在也要添加;
2、删除一个数,如果原本不存在则返回false,否则返回true;
3、随机返回一个数,每个数的返回的概率应该与其出现的频次成正比。
思路参考https://blog.csdn.net/qq_46105170/article/details/109269399。
思路是哈希表 + list。由于要 O ( 1 ) O(1) O(1)删除,我们需要存储每个数出现的下标。由于这里允许重复元素的存在,所以下标也要存多份。添加操作并不难,在删除的时候,避免错误的方法是,先判断要删除的数是否恰好等于list的最后一个数,如果是,那么很好删除;如果不是,那就确定了要删除的数和list最后一个数不一样,这时候就可以将本题看待成Duplicates not allowed的情形了,做哈希表的修改就不会出错。代码如下:
import java.util.*;
public class RandomizedCollection {
private List<Integer> list;
private Map<Integer, Set<Integer>> map;
private Random random;
/**
* Initialize your data structure here.
*/
public RandomizedCollection() {
list = new ArrayList<>();
map = new HashMap<>();
random = new Random();
}
/**
* Inserts a value to the collection. Returns true if the collection did not already contain the specified element.
*/
public boolean insert(int val) {
boolean res = false;
if (!map.containsKey(val) || map.get(val).isEmpty()) {
res = true;
}
list.add(val);
map.putIfAbsent(val, new HashSet<>());
map.get(val).add(list.size() - 1);
return res;
}
/**
* Removes a value from the collection. Returns true if the collection contained the specified element.
*/
public boolean remove(int val) {
if (!map.containsKey(val) || map.get(val).isEmpty()) {
return false;
}
Set<Integer> set = map.get(val);
if (val == list.get(list.size() - 1)) {
set.remove(list.size() - 1);
list.remove(list.size() - 1);
} else {
int pos = set.iterator().next();
set.remove(pos);
int last = list.get(list.size() - 1);
map.get(last).remove(list.size() - 1);
map.get(last).add(pos);
Collections.swap(list, pos, list.size() - 1);
list.remove(list.size() - 1);
}
return true;
}
/**
* Get a random element from the collection.
*/
public int getRandom() {
return list.get(random.nextInt(list.size()));
}
}
所有操作时间复杂度 O ( 1 ) O(1) O(1),空间 O ( n ) O(n) O(n)。