-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDay-12 Insert Delete GetRandom
55 lines (48 loc) · 1.53 KB
/
Day-12 Insert Delete GetRandom
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
class RandomizedSet {
/** Initialize your data structure here. */
ArrayList<Integer> num=null;
HashMap<Integer,Integer> map=null;
java.util.Random rand=new java.util.Random();
public RandomizedSet() {
map=new HashMap<>();
num=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(map.containsKey(val)){
return false;
}
else{
map.put(val,num.size());
num.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(map.containsKey(val)){
int index=map.get(val);
if(index<num.size()-1){
int last=num.get(num.size()-1);
num.set(index,last);
map.put(last,index);
}
num.remove(num.size()-1);
map.remove(val);
System.out.println(num);
return true;
}
return false;
}
/** Get a random element from the set. */
public int getRandom() {
return num.get(rand.nextInt(num.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();
*/