LeetCode 1588. 所有奇數長度子陣列的和 字首和
阿新 • • 發佈:2020-09-21
地址https://leetcode-cn.com/problems/sum-of-all-odd-length-subarrays/
給你一個正整數陣列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 <= 100 1 <= arr[i] <= 1000
演算法1
考慮到字首和 然後 如果起點終點之差為偶數也就是長度為奇數
累計相加即可
C++ 程式碼
class Solution { public: int a[150]; int sumOddLengthSubarrays(vector<int>& arr) { arr.insert(arr.begin(),0); for(int i = 1; i < arr.size();i++){ a[i] = a[i-1]+arr[i]; } int sum =0; for(int i = 1;i< arr.size();i++){ for(int j = i;j < arr.size();j++){ if( (j-i+1)%2 != 0 ){ sum+= a[j]-a[i-1]; } } } return sum; } };