forked from varunu28/LeetCode-Java-Solutions
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSliding Window Median.java
More file actions
72 lines (59 loc) · 1.63 KB
/
Sliding Window Median.java
File metadata and controls
72 lines (59 loc) · 1.63 KB
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
class Solution {
List<Long> list;
double[] ans;
boolean isEven;
public double[] medianSlidingWindow(int[] nums, int k) {
list = new ArrayList<>();
ans = new double[nums.length - k + 1];
isEven = k % 2 == 0;
for (int i=0; i<k; i++) {
list.add((long) nums[i]);
}
Collections.sort(list);
int idx = 0;
ans[idx++] = getMedian();
for (int i=k; i<nums.length; i++) {
removeElement(nums[idx - 1]);
int tempIdx = getBinarySearchIdx(nums[i]);
list.add(tempIdx, (long) nums[i]);
ans[idx++] = getMedian();
}
return ans;
}
private int getBinarySearchIdx(int num) {
if (list.size() == 0 || num < list.get(0)) {
return 0;
}
if (num > list.get(list.size() - 1)) {
return list.size();
}
int start = 0;
int end = list.size();
while (start < end) {
int mid = (start + end) / 2;
if (list.get(mid) < num) {
start = mid + 1;
}
else {
end = mid;
}
}
return end;
}
private void removeElement(int num) {
for (int i=0; i<list.size(); i++) {
if (list.get(i) == num) {
list.remove(i);
break;
}
}
}
private double getMedian() {
if (isEven) {
return (double) (list.get(list.size()/2) + list.get(list.size()/2 - 1)) / 2;
}
else {
return (double) list.get(list.size()/2);
}
}
}