给定一个元素都是正整数的数组A ,正整数 L 以及 R (L <= R)。
求连续、非空且其中最大元素满足大于等于L 小于等于R的子数组个数。
例如 :
输入:
A = [2, 1, 4, 3]
L = 2
R = 3
输出: 3
解释: 满足条件的子数组: [2], [2, 1], [3].
注意:
L, R 和 A[i] 都是整数,范围在 [0, 10^9]。
数组 A 的长度范围在[1, 50000]。
思路:我们将数组分段,其中的每一段都不包含非常大的元素(也就是大于R的元素),然后我们用每一段能组成的子数组数目减去最大值不在[L,R]之间的子数组数量即可。
class Solution {
public int numSubarrayBoundedMax(int[] A, int L, int R) {
int num1=0,num2=0;
long ans=0;
for(int i=0;i<A.length;i++) {
if(A[i]>=L && A[i]<=R) {
ans-=num1*(num1+1)/2;
num2++; num1=0;
}
else if(A[i]<L) {
num1++;
num2++;
}
else {
ans-=num1*(num1+1)/2;
ans+=num2*(num2+1)/2;
num1=0; num2=0;
}
}
ans+=num2*(num2+1)/2;
ans-=num1*(num1+1)/2;
return (int)ans;
}
}