这道题难度还是挺大的,基本上hard的题目都要费点脑子,根据一般的办法,求和并暴力的复杂度是O(N^2),而题目中明确说O(N^2)是naive,所以,该题目的难点在于如何降低时间复杂度。
题目大意:给定一组数和一个区间,求改组数中所有满足该区间加和的子数组的个数,就是从数组中挑选一组连续的数,如果选出的数之和满足条件,则视为一组有效数,如果不满足条件,则为无效,求所有有效数组的个数。
思路是,假设创建一个sums数组,sums[i]表示数组中前i个数据之和,挑选从j到i之间的数组,表示为sums[i]-sums[j],而该数组要满足lower <= sums[i]-sums[j] <= upper,sums[i]是我们当前遍历到的最远位置,sum[j]是已经遍历过的位置,所以,已经存在于数组中,而sum[i]是刚算出来的固定值,所以,只要在原来的sums数组中找到满足一定条件的sums[j],则j-i一定满足条件。原式是lower <= sums[i]-sums[j] <= upper,则sums[i]-upper <= sums[j] <= sums[i]-lower。
有了以上的分析思路就明确了,从0开始遍历,当遍历到数组中前i个值之和时,只要判断sums数组中,前i 个值中有几个满足sums[i]-upper <= sums[j] <= sums[i]-lower(j<i)即可,将结果不断累加,最终得到的就是所有满足条件的情况个数。
以下是AC代码:
class Solution {
public:
int countRangeSum(vector<int>& nums, int lower, int upper) {
int res = 0;
multiset<long long> numset;
numset.insert(0);
long long sum = 0;
for (int i=0; i<nums.size(); ++i) {
sum += nums[i];
res += distance(numset.lower_bound(sum-upper), numset.upper_bound(sum-lower));
numset.insert(sum);
}
return res;
}
};