-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathInsert_Delete_GetRandom.java
57 lines (41 loc) · 1.13 KB
/
Insert_Delete_GetRandom.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
package AllCodes;
import java.util.*;
class RandomizedSet {
private Map<Integer,Integer>map;
private ArrayList<Integer>list;
private Random r;
public RandomizedSet() {
map=new HashMap<>();
list=new ArrayList<>();
r=new Random();
}
public boolean insert(int val) {
if(map.containsKey(val)){
return false;
}
map.put(val,list.size());
list.add(val);
return true;
}
public boolean remove(int val) {
if(!map.containsKey(val)){
return false;
}
int removeIndex=map.get(val);
map.remove(val);
if(removeIndex==list.size()-1){
list.remove(removeIndex);
return true;
}
list.set(removeIndex,list.get(list.size()-1));
list.remove(list.size()-1);
map.put(list.get(removeIndex), removeIndex );
return true;
}
public int getRandom() {
int i = r.nextInt(list.size());
return list.get(i);
}
}
public class Insert_Delete_GetRandom {
}