327. count-of-range-sum

37 篇文章 0 订阅
12 篇文章 0 订阅

         这道题难度还是挺大的,基本上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;
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值