the first one in Biweekly Contest 35.
Difficulty : Easy
Related Topics : Array
Given an array of positive integers
arr
, calculate the sum of all possible odd-length subarrays.A subarray is a contiguous subsequence of the array.
Return the sum of all odd-length subarrays of
arr
.Input: arr = [1,4,2,5,3] Output: 58 Explanation: The odd-length subarrays of arr and their sums are: [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 If we add all these together we get 1 + 4 + 2 + 5 + 3 + 7 + 11 + 10 + 15 = 58
Input: arr = [1,2] Output: 3 Explanation: There are only 2 subarrays of odd length, [1] and [2]. Their sum is 3.
Input: arr = [10,11,12] Output: 66
1 <= arr.length <= 100
1 <= arr[i] <= 1000
- mine
- Java
Runtime: 1 ms, faster than 88.25%, Memory Usage: 37.3 MB, less than 68.44% of Java online submissions
// O(N^2)time // O(N)space public int sumOddLengthSubarrays(int[] arr) { int res = 0; int n = arr.length; int[] sum = new int[n + 1]; for(int i = 0; i < n; i++){ sum[i + 1] = sum[i] + arr[i]; } for (int i = 0; i < n; i++){ for (int j = 1; i + j <= n; j += 2) { res += sum[i + j] - sum[i]; } } return res; }
- Java