03、数组中达标子数组的数量问题

/**
 * 暴力解法如下
 *
 * @param arr
 * @param num
 * @return
 */
public static int getNumBao(int[] arr, int num) {
    if (arr == null || arr.length == 0) {
        return 0;
    }
    //达标子数组的数量
    int valid = 0;
    for (int L = 0; L < arr.length; L++) {
        for (int R = L; R < arr.length; R++) {
            //当前这个子数组中的最大值和最小值
            int max = 0;
            int min = 0;
            for (int i = L; i <= R; i++) {
                if (arr[i] > max) {
                    max = arr[i];
                }
                if (arr[i] < min) {
                    min = arr[i];
                }
            }
            //sub最大值 - sub中最小值 <= num
            //说明这个子数组符合要求，数量++
            if (max - min <= num) {
                valid++;
            }
        }
    }
    return valid;
}

 

/**
 * 利用滑动窗口解法如下：
 * 时间复杂度O(N)
 *
 * @param arr 数组
 * @param num 固定值
 * @return
 */
public static int getNum(int[] arr, int num) {
    if (arr == null || arr.length == 0) {
        return 0;
    }
    //队列元素是从小到大（1、2、3、4）
    LinkedList<Integer> qmin = new LinkedList<Integer>();
    //队列元素是从大到小的（4、3、2、1）
    LinkedList<Integer> qmax = new LinkedList<Integer>();
    int L = 0;
    int R = 0;
    int res = 0;
    while (L < arr.length) { // L是开头位置，尝试每一个开头
        // 如果此时窗口的开头是L,下面的while工作:R向右扩到违规为止
        while (R < arr.length) { // R是最后一个达标位置的再下一个
            while (!qmin.isEmpty() && arr[qmin.peekLast()] >= arr[R]) {
                qmin.pollLast();
            }
            qmin.addLast(R);
            // R -> arr[R],
            while (!qmax.isEmpty() && arr[qmax.peekLast()] <= arr[R]) {
                qmax.pollLast();
            }
            qmax.addLast(R);
            //如果当前窗口的最大值减去最小值大于num了，说明已经达到了违规的位置，直接停下
            if (arr[qmax.getFirst()] - arr[qmin.getFirst()] > num) {
                break;
            }
            R++;
        }
        // R是最后一个达标位置的再下一个，第一个违规的位置
        res += R - L;
        if (qmin.peekFirst() == L) {
            qmin.pollFirst();
        }
        if (qmax.peekFirst() == L) {
            qmax.pollFirst();
        }
        L++;
    }
    return res;
}