前缀和算法(所有奇数长度子数组的和)

文章目录

题目

给你一个正整数数组 arr ,请你计算所有可能的奇数长度子数组的和。

子数组 定义为原数组中的一个连续子序列。

请你返回 arr 中 所有奇数长度子数组的和 。

前缀和

假设我们有一个整型数组arr,另有一个数组preSum存储该数组的前缀和,也就是说 p r e S u m [ j ] = ∑ i = 0 j − 1 a r r [ i ] preSum[j]=\sum_{i=0}^{j-1}arr[i] preSum[j]=i=0j1arr[i]
我们记 p r e S u m [ 0 ] = 0 preSum[0]=0 preSum[0]=0,另外preSum的长度比arr多1。

也就是说preSum[i]表示arr[0]到arr[i-1]的和。

如此一来,我们在计算区间 a r r [ i : j ] arr[i:j] arr[i:j]的和时,只需要 p r e S u m [ j + 1 ] − p r e S u m [ i ] preSum[j+1]-preSum[i] preSum[j+1]preSum[i]即可。

对于这样一道题来说,减少了每次循环计算子数组的和的时间。

 int sumOddLengthSubarrays(vector<int>& arr) {
        int n = arr.size();
        vector<int> prefixSums(n + 1);
        for (int i = 0; i < n; i++) {
            prefixSums[i + 1] = prefixSums[i] + arr[i];
        }
        int sum = 0;
        for (int start = 0; start < n; start++) {
            for (int length = 1; start + length <= n; length += 2) {
                int end = start + length - 1;
                sum += prefixSums[end + 1] - prefixSums[start];
            }
        }
        return sum;
    }

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/sum-of-all-odd-length-subarrays/solution/suo-you-qi-shu-chang-du-zi-shu-zu-de-he-yoaqu/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

右边是我女神

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值