-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSlidingWindowMaxmium.java
More file actions
81 lines (73 loc) · 2.6 KB
/
Copy pathSlidingWindowMaxmium.java
File metadata and controls
81 lines (73 loc) · 2.6 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
73
74
75
76
77
78
79
80
81
package com.al;
import org.junit.jupiter.api.Test;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Map;
import java.util.PriorityQueue;
public class SlidingWindowMaxmium {
@Test
void test() {
Solution s = new Solution();
int[] ans = s.maxSlidingWindow(new int[]{1,3,-1,-3,5,3,6,7},3);
for (int i = 0; i < ans.length; i++) {
System.out.print(ans[i] + ",");
}
System.out.println();
ans = s.maxSlidingWindow(new int[]{1},1);
for (int i = 0; i < ans.length; i++) {
System.out.print(ans[i] + ",");
}
System.out.println();
ans = s.maxSlidingWindow(new int[]{1,-1},1);
for (int i = 0; i < ans.length; i++) {
System.out.print(ans[i] + ",");
}
System.out.println();
ans = s.maxSlidingWindow(new int[]{9,11},2);
for (int i = 0; i < ans.length; i++) {
System.out.print(ans[i] + ",");
}
System.out.println();
ans = s.maxSlidingWindow(new int[]{4,-2},2);
for (int i = 0; i < ans.length; i++) {
System.out.print(ans[i] + ",");
}
}
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
// set compare function
PriorityQueue<Map> p = new PriorityQueue(k, new Comparator<Map<Integer, Integer>>() {
@Override
public int compare(Map o1, Map o2) {
return (Integer) o2.keySet().iterator().next() - (Integer) o1.keySet().iterator().next();
}
});
// the first window
for (int i = 0; i < k; i++) {
HashMap map = new HashMap();
map.put(nums[i],i);
p.offer(map);
}
int n = nums.length;
int[] ans = new int[n-k+1];
// get the first maximum
ans[0] = (int) p.peek().keySet().iterator().next();
// [1,3,-1,-3,5,3,6,7], 3
// [3,-1,-3]
// [-1.-3,5]
for (int i = k; i < n; i++) {
HashMap map = new HashMap();
map.put(nums[i],i);
// insert into the priority queue
p.offer(map);
// the maximum value may not in the current window
// if not exist, poll the value
while ((Integer)(p.peek().values().iterator().next()) <= (i-k)) {
p.poll();
}
ans[i-k+1] = (int) p.peek().keySet().iterator().next();
}
return ans;
}
}
}