1.8最大值减去最小值小于或等于num的子数组数量

题目

给定数组arr和整数num，共返回有多少个子数组满足如下情况：

max(arr[i…j]) - min(arr[i…j]) <= num

代码实现

public int getNum(int[] arr, int num) {
    if (arr == null || arr.length == 0) {
        return 0;
    }

    LinkedList<Integer> qmin = new LinkedList<>(); //维护窗口子数组的最大值更新
    LinkedList<Integer> qmax = new LinkedList<>(); //维护窗口子数组的最小值更新
    int i = 0;
    int j = 0;
    int res = 0;

    while (i < arr.length) {
        while (j < arr.length) {
            while (!qmin.isEmpty() && arr[qmin.peekLast()] >= arr[j]) {
                qmin.pollLast();
            }
            qmin.addLast(j);

            while (!qmax.isEmpty() && arr[qmax.peekLast()] <= arr[j]) {
                qmax.pollLast();
            }
            qmax.addLast(j);

            if (arr[qmax.getFirst()] - arr[qmin.getFirst()] > num) {
                break;
            }
            j++;
        }
        if (qmin.peekFirst() == i) {
            qmin.pollFirst();
        }
        if (qmax.peekFirst() == i) {
            qmax.pollFirst();
        }
        res += j - i; //以arr[i]作为第一个元素的子数组个数
        i++;
    }

    return res;
}