最大值减去最小值小于或等于num的连续子数组数量（offer收割）

题目

给定数组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 com.iqiyi;

import java.util.LinkedList;

public class Code1_10 {
	public static int getNum(int[] arr,int num){
		int ans=0;
		int i=0;
		int j=0;
		LinkedList<Integer> linkedList1=new LinkedList<Integer>();
		LinkedList<Integer> linkedList2=new LinkedList<Integer>();
		while(i<arr.length){
			while(j<arr.length){
				while(!linkedList1.isEmpty()&&arr[linkedList1.peekLast()]<arr[j])
					linkedList1.removeLast();
				linkedList1.addLast(j);
				while(!linkedList2.isEmpty()&&arr[linkedList2.peekLast()]>arr[j])
					linkedList2.removeLast();
				linkedList2.addLast(j);
				if((arr[linkedList1.peekFirst()]-arr[linkedList2.peekFirst()])>num){
					break;
				}
				j++;
			}
			ans+=j-i;
			if(linkedList1.peekFirst()==i)
				linkedList1.removeFirst();
			if(linkedList2.peekFirst()==i)
				linkedList2.removeFirst();
			i++;
		}
		return ans;
	}
	
	public static void main(String[] args) {
		System.out.println(getNum(new int[]{2,3,1,6,4}, 2));
	}
}
复制代码

转载于:https://juejin.im/post/5c3d533b6fb9a04a0b226ce6