-
Notifications
You must be signed in to change notification settings - Fork 2
/
Insert-Delete-GetRandom-O(1).java
45 lines (41 loc) · 1.27 KB
/
Insert-Delete-GetRandom-O(1).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
class RandomizedSet {
private Map<Integer,Integer> keyVals;
private List<Integer> vals;
private Random r;
public RandomizedSet() {
keyVals = new HashMap<>();
vals = new ArrayList<>();
r = new Random();
}
public boolean insert(int val) {
if(keyVals.containsKey(val)){
return false;
}
keyVals.put(val,vals.size());
vals.add(val);
return true;
}
public boolean remove(int val) {
if(!keyVals.containsKey(val)){
return false;
}
int delPair = keyVals.get(val);
int delVal = vals.get(vals.size()-1);
vals.set(delPair,delVal); // element moved to last place
keyVals.put(delVal,delPair); // moved element index updated
vals.remove(vals.size()-1); // now remove last element
keyVals.remove(val); // remove map pair
return true;
}
public int getRandom() {
int rand_pos = r.nextInt(vals.size());
return vals.get(rand_pos);
}
}
/**
* 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();
*/