-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathQueueDay2.java
More file actions
85 lines (69 loc) · 2.07 KB
/
Copy pathQueueDay2.java
File metadata and controls
85 lines (69 loc) · 2.07 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
82
83
84
85
package queue;
import java.util.ArrayDeque;
import java.util.Arrays;
public class QueueDay2 {
public static void main(String[] args) {
// int[] arr = {2, 5, 1, 3, 2, 1, 4, 5};
// int[] ans = slidingWindowMaximum(arr, 4);
// System.out.println(Arrays.toString(ans)); // [5,5,3,4,5]
int[] arr = {12, -1, -7, 8, -15, 30, 16, 28};
int[] ans = firstNegativeNumberInWindow(arr, 3);
System.out.println(Arrays.toString(ans));
}
static int[] slidingWindowMaximum(int[] arr, int k) {
ArrayDeque<Integer> dq = new ArrayDeque<>();
int i = 0;
while (i < k) {
if (!dq.isEmpty() && dq.getLast() < arr[i]) {
dq.pollLast();
}
dq.offerLast(arr[i]);
i++;
}
int n = arr.length;
int ans[] = new int[n - k + 1];
for (; i < n; i++) {
ans[i - k] = dq.peekFirst();
if (arr[i - k] == dq.peekFirst()) {
dq.pollFirst();
}
while (!dq.isEmpty() && dq.getLast() < arr[i]) {
dq.pollLast();
}
dq.offerLast(arr[i]);
}
ans[n - k] = dq.peekFirst();
return ans;
}
static int[] firstNegativeNumberInWindow(int arr[], int k) {
ArrayDeque<Integer> dq = new ArrayDeque<>();
int i = 0;
while (i < k) {
if (arr[i] < 0) {
dq.offerLast(arr[i]);
}
i++;
}
int n = arr.length;
int[] ans = new int[n - k + 1];
for (; i < n; i++) {
if (dq.isEmpty()) {
ans[i - k] = 0;
} else {
ans[i - k] = dq.peekFirst();
}
if (!dq.isEmpty() && arr[i - k] == dq.peekFirst()) {
dq.pollFirst();
}
if (arr[i] < 0) {
dq.offerLast(arr[i]);
}
}
if (dq.isEmpty()) {
ans[n - k] = 0;
} else {
ans[n - k] = dq.peekFirst();
}
return ans;
}
}