【Leetcode】1524. Number of Sub-arrays With Odd Sum

题目地址：

https://leetcode.com/problems/number-of-sub-arrays-with-odd-sum/

给定一个长$n$的数组$A$，问其和为奇数的非空子数组的个数。结果模$10^9+7$后返回。

先求前缀和数组$c[i]$即$c[i]={\sum }_{k=0}^{i-1}A[k]$，再用一个哈希表记录偶数和奇数前缀和的数量。接着遍历$A$，当遍历到$A[i]$的时候，如果前缀和$c[i+1]$是奇数，则查一下$c[0:i]$里有多少个偶数，累加之，就求出了以$A[i]$为右端点的和为奇数的子数组个数；如果前缀和$c[i+1]$是偶数也类似做。代码如下：

public class Solution {
    public int numOfSubarrays(int[] arr) {
        int MOD = (int) (1E9 + 7);
        // 为了加速，可以用数组代替哈希表，count[0]代表偶数个数，[1]代表奇数个数；
        // 空数组的前缀和是0，要计入
        int[] count = {1, 0};
        
        int res = 0;
        // 为了节省空间，不需要另外开数组算前缀和，可以直接用一个变量记录
        for (int i = 0, rem = 0; i < arr.length; i++) {
            rem = (rem + arr[i]) % 2;
            res = (res + count[rem ^ 1]) % MOD;
            count[rem]++;
        }
        
        return res;
    }
}

时间复杂度$O(n)$，空间$O(1)$。