-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathsolution.java
More file actions
40 lines (32 loc) · 1.43 KB
/
Copy pathsolution.java
File metadata and controls
40 lines (32 loc) · 1.43 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
import java.util.*;
class Solution {
public ArrayList<Integer> longestSubarray(int[] arr, int x) {
int n = arr.length;
ArrayList<Integer> result = new ArrayList<>();
if (n == 0) return result;
Deque<Integer> maxD = new ArrayDeque<>(); // indices, values decreasing
Deque<Integer> minD = new ArrayDeque<>(); // indices, values increasing
int l = 0, bestL = 0, bestLen = 0;
for (int r = 0; r < n; r++) {
// maintain max deque (decreasing)
while (!maxD.isEmpty() && arr[r] > arr[maxD.peekLast()]) maxD.pollLast();
maxD.offerLast(r);
// maintain min deque (increasing)
while (!minD.isEmpty() && arr[r] < arr[minD.peekLast()]) minD.pollLast();
minD.offerLast(r);
// shrink window if invalid
while (!maxD.isEmpty() && !minD.isEmpty() && (long)arr[maxD.peekFirst()] - arr[minD.peekFirst()] > x) {
if (maxD.peekFirst() == l) maxD.pollFirst();
if (minD.peekFirst() == l) minD.pollFirst();
l++;
}
int len = r - l + 1;
if (len > bestLen) { // strictly greater to prefer smaller start index on ties
bestLen = len;
bestL = l;
}
}
for (int i = bestL; i < bestL + bestLen; i++) result.add(arr[i]);
return result;
}
}