package com.my.suanfa;
import java.util.LinkedList;
/**
* 最大值减去最小值小于或等于num的子数组数量
* 使用两个双端队列来统计每个子数组中的最大值和最小值
* 时间复杂度为O(N),额外空间复杂度为O(N)
* */
public class Solution06 {
public int getNum(int[] arr, int num) {
//边界条件判断
if(arr == null || arr.length == 0) {
return 0;
}
//定义变量i表示子数组的第一个元素的下标
int i = 0;
//定义变量j表示子数组的最后一个元素的下标
int j = 0;
//定义变量res记录满足条件的子数组个数
int res = 0;
//定义双端队列qmax记录子数组中的最大值
LinkedList<Integer> qmax = new LinkedList<Integer>();
//定义双端队列qmin记录子数组中的最小值
LinkedList<Integer> qmin = new LinkedList<Integer>();
//循环遍历找到子数组的最大值和最小值,并统计满足要求的子数组的个数
while(i < arr.length) {
while(j < arr.length) {
while(!qmax.isEmpty() && arr[qmax.peekLast()] <= arr[j]) {
//如果qmax不为空,并且队尾元素小于等于正在遍历的元素,说明此刻队尾的元素并不是最大值,便将队尾元素出队
qmax.pollLast();
}
qmax.addLast(j);
while(!qmin.isEmpty() && arr[qmin.peekLast()] >= arr[j]) {
//如果qmin不为空,并且队尾元素大于等于正在遍历的元素,说明此刻队尾元素并不是最小值,便将队尾元素出队
qmin.pollLast();
}
qmin.addLast(j);
//不满足条件则此后所有元素都不满足条件,便跳出循环
if(arr[qmax.peekFirst()] - arr[qmin.peekFirst()] > num) {
break;
}
j++;
}
//因为j已经遍历完成,此时i会自增,如果i对应的元素还在队列中,则已经失效,应该将i对应的元素移除队列,因为i的下标最小,因此只可能是队列的第一个元素
if(qmax.peekFirst() == i) {
qmax.pollFirst();
}
//同理
if(qmin.peekFirst() == i) {
qmin.pollFirst();
}
res+= j - i;
i++;
}
return res;
}
}
010-最大值减去最小值小于或等于num的子数组数量
最新推荐文章于 2022-12-08 16:14:08 发布