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

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

给定数组 arr 和整数 num,共返回有多少个子数组满足如下情况:
max(arr[i…j]) - min(arr[i…j]) <= num
max(arr[i…j])表示子数组 arr[i…j]中的最大值,min(arr[i…j])表示子数组 arr[i…j]中的最 小值。
如果数组长度为 N,请实现时间复杂度为 O(N)的解法。
这道题类似于求滑动窗口最大值,可以参考昨天写的这个.滑动窗口最大值
代码如下(有注释):

package Main;

import java.util.LinkedList;

public class Main {

	public static 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;  //i~j的范围 
		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) {    //<=num时不跳出，j++继续向后查找
					break;          
				}
				j++;
			}
			/*
			 * 说明此时的最大值减去最小值>num了，要记录此时的i~j的范围中共有多少子数组
			 */
			if (qmin.peekFirst() == i) {  
 				qmin.pollFirst();
			}
			if (qmax.peekFirst() == i) {
				qmax.pollFirst();
			}
			res +=j -i;  //i~j-1及其子数组
			i++; //i向后移动
		}
		return res;
	}
}