这道题的两个队列都是严格单调队列,可以结合之前的单调栈结构和生成窗口最大值数组两道题结合起来理解。不过,为什么 1, 2, 1 num = 1 时,结果是 6 呢?
需要注意的几点是:
- 双端队列的队尾对应窗口右边界,双端队列的对头对应窗口的左边界,一般先操作右边界,不满足窗口条件了,再操作左边界。
- 第20行,是为了避免重复计算,因为break之后,j 就不会 ++ 了
import java.util.LinkedList;
import java.io.IOException;
import java.io.BufferedReader;
import java.io.InputStreamReader;
public class Main {
public static int getNum(int[] array, int num) {
if (array == null || array.length == 0 || num < 0) {
return 0;
}
LinkedList<Integer> qmax = new LinkedList<>();
LinkedList<Integer> qmin = new LinkedList<>();
int i = 0;
int j = 0;
int res = 0;
int len = array.length;
while (i < len) {
while (j < len) {
// 避免重复计算,因为break之后,j 就不会 ++ 了
if (qmin.isEmpty() || qmin.peekLast() != j) {
while (!qmin.isEmpty() && array[qmin.peekLast()] >= array[j]) {
qmin.pollLast();
}
qmin.addLast(j);
while (!qmax.isEmpty() && array[qmax.peekLast()] <= array[j]) {
qmax.pollLast();
}
qmax.addLast(j);
}
if (array[qmax.peekFirst()] - array[qmin.peekFirst()] > num) {
break;
}
++j;
}
res += j - i;
if (qmax.peekFirst() == i) {
qmax.pollFirst();
}
if (qmin.peekFirst() == i) {
qmin.pollFirst();
}
++i;
}
return res;
}
public static void main(String[] args) throws IOException {
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
String s = "";
while ((s = bf.readLine()) != null) {
String[] para = s.split(" ");
int len = Integer.parseInt(para[0]);
int num = Integer.parseInt(para[1]);
int[] arr = new int[len];
String[] elements = bf.readLine().split(" ");
for (int i = 0; i < len; i++) {
arr[i] = Integer.parseInt(elements[i]);
}
System.out.println(getNum(arr, num));
}
}
}