【数据结构-前缀&&哈希表】力扣1524. 和为奇数的子数组数目

给你一个整数数组 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

初始代码（超时）

class Solution {
public:
    int numOfSubarrays(vector<int>& arr) {
        int count = 0;
        vector<int> group;
        for(int i = 0;i < arr.size();i++){
            count = 0;
            for(int j = i;j < arr.size();j++){
                count += arr[j];
                if(count % 2 == 1){
                    group.push_back(count);
                }
            }
        }
        return group.size();
    }
};

时间复杂度O(n^2)
空间复杂度O(n^2)

第一次通过计算每个子段的和，如果是奇数就进行返回，但是会发现运算量过大。所以有没有一种办法可以不用对每个子段的和进行运算？

优化后的代码

class Solution {
public:
    int numOfSubarrays(vector<int>& arr) {
        const int MOD = 1e9 + 7;
        int sum = 0;
        int sub = 0;
        int odd = 0, even = 1;
        for(int i = 0;i < arr.size();i++){
            sum += arr[i];
            sub = (sub + (sum % 2 == 0 ? odd : even)) % MOD;;
            if(sum % 2 == 0){
                even++;
            }else{
                odd++;
            }
        }
        return sub;
    }
};

时间复杂度：O(n)，其中 n 是数组 arr 的长度。遍历数组一次，对于数组中的每个元素，更新前缀和、和为奇数的子数组数目以及 odd 和 even 的值的时间复杂度都是 O(1)，因此总时间复杂度是 O(n)。
空间复杂度：O(1)。只需要维护常量的额外空间。

可以使用前缀和来做，并且由于我们只需要找到奇数子段和的数量，不需要计算他们。仔细思考会发现，当一个奇数减去一个偶数的时候，结果还是奇数，当一个偶数减去一个奇数的时候，结果是奇数。在前缀和中，两个前缀和相减就是他们两个位置之间的子段和。所以我们在计算前缀和的时候，记前面有odd个奇数前缀和，even个偶数前缀和。当前前缀和为偶数的时候，在这位置及之前就有odd个奇数子段和，当前前缀和为奇数的时候，在这位置及之前就有even个奇数子段和。