这道题给了我们一个数组,又给了我们一个下限和一个上限,让我们求有多少个不同的区间使得每个区间的和在给定的上下限之间。
这道题要求不能用O(n^2)的方法。看到i到j的范围的和,就得先想到累积和!有了累积和,我们就能在O(1)的时间范围内得到区间[i,j]的和。但是得到区间[i,j]的和还不够,总不能遍历所有[i,j]的组合来看满足条件的区间吧,这样不就又成了O(n^2)了。方法是二分查找,由于lower <= sum(i,j) <= upper,即lower <= sums[j] - sums[i] <= upper,所以sums[j] - upper <= sums[i] <= sums[j] - lower。我们把之前得到的前i-1项的累积和都放到一个排序的数组中,如multiset中,这样前面求出来的累积和都是排好序的。sums[i]求出来之后,我们需要看哪些i可以构成满足要求的[i,j]组合。也就是哪些i能满足lower <= sums[j] - sums[i] <= upper。可以利用lower_bound和upper_bound函数找到i前面两个边界m和n,那么m到n之间都满足。
class Solution {
public:
int countRangeSum(vector<int>& nums, int lower, int upper) {
int res = 0;
long long sum = 0;
multiset<long long> sums;
sums.insert(0);
for(int i = 0; i < nums.size(); ++i){
sum += nums[i];
res += distance(sums.lower_bound(sum - upper), sums.upper_bound(sum - lower));
sums.insert(sum);
}
return res;
}
};