将问题转化为求最大值不超过right的子数组的个数减去最大值不超过left-1的个数,就是最大值范围在left和right之间子数组的个数。
而最大值不超过k的子数组的个数,可以通过双指针分段很好的解决。
using LL = long long;
class Solution {
public:
int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
auto numSubarrayLessK = [&](int k) -> LL {
LL ans = 0;
for (int i = 0; i < nums.size(); i++) {
if (nums[i] > k) continue;
int j = i + 1;
while (j < nums.size() && nums[j] <= k) j++;
int t = j - i;
ans += (LL)t * (t + 1) / 2;
i = j;
}
return ans;
};
return numSubarrayLessK(right) - numSubarrayLessK(left - 1);
}
};