【题目】
给定数组 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]中的最 小值。
package orc.shi.s;
import java.util.LinkedList;
public class test1 {
public static void main(String[] args) {
//int[] arr = getRandomArray(10);
int[] arr= {3, 1 ,5 ,800 ,8, 7, 0};
int num = 5;
printArray(arr);
System.out.println("结果"+getNum(arr, num));
System.out.println("结果"+getNum1(arr, num));
}
public static void printArray(int[] arr) {
if (arr != null) {
for (int i = 0; i < arr.length; i++) {
System.out.print(arr[i] + " ");
}
System.out.println();
}
}
public static int[] getRandomArray(int len) {
if (len < 0) {
return null;
}
int[] arr = new int[len];
for (int i = 0; i < len; i++) {
arr[i] = (int) (Math.random() * 10);
}
return arr;
}
public static int getNum(int[] arr,int num) {
int res=0;
int i=0;
int j=0;
LinkedList<Integer> min=new LinkedList();//维护窗口最小值
LinkedList<Integer> max=new LinkedList();//维护窗口最大值
for( i=0;i<arr.length;i++) {
while(j<arr.length) {
//维护max
while(!max.isEmpty()&&arr[j]>=arr[max.peekLast()]) {
/**
* 这里也太容易出错了,我靠 ,真的太容易出错了,我醉了,从头跟了几次才发现
* 就是如果j加进来之后不满足,那么还会进j,如果还不行,i继续移动,又一次还是j
* 这样如果没有等于号,那么j不断的加到max里面了,这样,如果可以的时候,即使删除第一个
* j,但是后面还有好几个呢。。这样就会导致有些没有算到。,所以不管什么时候,只要是滑动窗口的问题,带个等号就完事了。
*/
max.pollLast();
}
max.addLast(j);
//维护min
while(!min.isEmpty()&&arr[j]<=arr[min.peekLast()]) {
min.pollLast();
}
min.addLast(j);
//向右滑动维护结束
//判断此时是否合理?
if((arr[max.peek()]-arr[min.peek()])>num) {//这里也要注意,本题可以,换个题可能也要加限制
//i滑动一个位置
if(i==min.peekFirst()) {
min.pollFirst();
}
if(i==max.peekFirst()) {
max.pollFirst();
}
break;
}
j++;
}
res+=j-i;
}
return res;
}
public static int getNum1(int[] arr, int num) {
if (arr == null || arr.length == 0) {
return 0;
}
LinkedList<Integer> qmin = new LinkedList<Integer>();
LinkedList<Integer> qmax = new LinkedList<Integer>();
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;
i++;
}
return res;
}
}