1. 程式人生 > 實用技巧 >5457. 和為奇數的子陣列數目。 字首和

5457. 和為奇數的子陣列數目。 字首和

給你一個整數陣列 arr 。請你返回和為 奇數 的子陣列數目。

由於答案可能會很大,請你將結果對 10^9 + 7 取餘後返回。

示例 1:

輸入:arr = [1,3,5]
輸出:4
解釋:所有的子陣列為 [[1],[1,3],[1,3,5],[3],[3,5],[5]] 。
所有子陣列的和為 [1,4,9,3,8,5].
奇數和包括 [1,9,3,5] ,所以答案為 4 。
示例 2 :

輸入:arr = [2,4,6]
輸出:0
解釋:所有子陣列為 [[2],[2,4],[2,4,6],[4],[4,6],[6]] 。
所有子陣列和為 [2,6,12,4,10,6] 。
所有子陣列和都是偶數,所以答案為 0 。
示例 3:

輸入:arr = [1,2,3,4,5,6,7]
輸出:16
示例 4:

輸入:arr = [100,100,99,99]
輸出:4
示例 5:

輸入:arr = [7]
輸出:1

提示:

1 <= arr.length <= 10^5
1 <= arr[i] <= 100

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/number-of-sub-arrays-with-odd-sum
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

class Solution {
public:
    int numOfSubarrays(vector<int>& arr) {
        //空陣列的話為0,所以even初始為1
        int odd = 0, even = 1;
        long long ans = 0;
        int sum = 0;

        for (int num : arr) {
            sum += num;
            //若當前字首和為偶,去掉裡面的奇數才是奇,所以加上此時奇數個數
            //若當前字首和為奇,去掉裡面的偶數也還是奇,所以加上此時偶數個數
            ans += (sum % 2 == 0 ? odd : even);
            if (sum % 2 == 0) {
                ++even;
            }
            else {
                ++odd;
            }
        }

        return ans % 1000000007;
    }
};