class Solution {
public:
int sumOddLengthSubarrays(vector<int>& arr) {
}
};
1588. 所有奇数长度子数组的和
给你一个正整数数组 arr ，请你计算所有可能的奇数长度子数组的和。
子数组 定义为原数组中的一个连续子序列。
请你返回 arr 中 所有奇数长度子数组的和 。
示例 1：
输入：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
示例 2：
输入：arr = [1,2] 输出：3 解释：总共只有 2 个长度为奇数的子数组，[1] 和 [2]。它们的和为 3 。
示例 3：
输入：arr = [10,11,12] 输出：66
提示：
1 <= arr.length <= 1001 <= arr[i] <= 1000
进阶：
你可以设计一个 O(n) 时间复杂度的算法解决此问题吗？
原站题解
golang 解法, 执行用时: 0 ms, 内存消耗: 2.2 MB, 提交时间: 2020-11-17 07:16:47
func sumOddLengthSubarrays(arr []int) int {
n, res := len(arr), 0
for i := 0; i < n; i++ {
left, right := i + 1, n - i
left_even, right_even := (left + 1) / 2, (right + 1) / 2
left_odd, right_odd := left / 2, right / 2
res += (left_even * right_even + left_odd * right_odd) * arr[i]
}
return res
}
python3 解法, 执行用时: 36 ms, 内存消耗: 13.5 MB, 提交时间: 2020-11-16 20:41:11
class Solution:
def sumOddLengthSubarrays(self, arr: List[int]) -> int:
n, res = len(arr), 0
for i in range(0, n):
left, right = i + 1, n - i
left_even, right_even = (left + 1) // 2, (right + 1) // 2
left_odd, right_odd = left // 2, right // 2
res += (left_even * right_even + left_odd * right_odd) * arr[i]
return res