给定一个元素都是正整数的数组A ,正整数 L 以及 R (L <= R)。
求连续、非空且其中最大元素满足大于等于L 小于等于R的子数组个数。
例如 :
输入:
A = [2, 1, 4, 3]
L = 2
R = 3
输出: 3
解释: 满足条件的子数组: [2], [2, 1], [3].
查找满足小于等于R,大于等于L的字数组可以转化为:小于等于R的字数组 - 小于L的字数组,for循环扫两遍即可
class Solution {
public:
int numSubarrayBoundedMax(vector<int>& A, int L, int R) {
int s1 = 0, s2 = 0;
for(int i = 0, j = 0; i < A.size(); ) {
if(A[i] <= R) {
i++;
s1 += i - j;
}
else {
i++;
j = i;
}
}
for(int i = 0, j = 0; i < A.size(); ) {
if(A[i] < L) {
i++;
s2 += i - j;
}
else {
i++;
j = i;
}
}
return s1 - s2;
}
};