【题目描述】
给你一个正整数数组 arr ,请你计算所有可能的奇数长度子数组的和。子数组 定义为原数组中的一个连续子序列。请你返回 arr 中 所有奇数长度子数组的和 。
示例:
输入:arr = [1,4,2,5,3]
输出:58
解释:所有奇数长度子数组和它们的和为:
[1] = 1
[4] = 4
[2] = 2
[5] = 5
[3] = 3
[1,4,2] = 7
[4,2,5] = 11
[2,5,3] = 10
[1,4,2,5,3] = 15
我们将所有值求和得到 1 + 4 + 2 + 5 + 3 + 7 + 11 + 10 + 15 = 58
解法一
解法二
思路:
我们只需要枚举所有的奇数长度的子数组即可。在下面的代码中,i 用来枚举每个连续子数组的起点,sz 为连续子数组的长度。accumulate用来计算起点是 i,长度为 sz 的子数组的和。时间复杂度是 O(n^3)
解法三
思路:
在上面的解法二中,accumulate的本质就是在计算从 arr[i] 到 arr[i + sz - 1] 的连续子数组的和。计算连续子数组的和,很容想到可以使用前缀和(Prefix Sum)的方式。使用 O(n) 的时间可以预处理前缀和数组,之后使用 O(1) 的时间即可计算出一个连续子数组的和。
时间复杂度是 O(n^2) 的,空间复杂度是 O(n) 的。
解法四
思路:
- odd奇数,even偶数
- 对于每个元素i(数组中下标为i)来说,要构成奇数长度的子数组
即 i左边的元素个数left+i本身自己一个+右边元素的个数right=奇数
即 left+right=偶数- 满足a+b=偶数就只有两种情况
- 奇数+奇数=偶数
- 偶数+偶数=偶数
- 所以只需要求得i左边可以选择奇数长度的可能有多少种,即left_odd,同样求右边奇数right_odd就可以求出策略1有多少种可能
- 所以只需要求得i左边可以选择偶数长度的可能有多少种,即left_odd,同样求右边偶数right_odd
就可以求出策略1有多少种可能,注意0也算选择的一种可能
即元素i在所有奇数长度子数组出现的次数总和是
left_odd * right_odd+left_even * right_even
- 元素i左边元素共有i + 1个元素个,右边元素共有siz-i个(包括元素自身)