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

文章目录

  • 题目
  • 前缀和

题目

给你一个正整数数组 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)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

你可能感兴趣的:(算法竞赛入门,算法,leetcode,数据结构)