LeetCode 1524. 和为奇数的子数组数目--前缀和取模

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

提示：

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

题解

前缀和sum[x]为奇数，前缀和sum[y]为偶数，那么区间和[x+1,y]为奇数，所以针对每个位置的前缀和，如果为奇数判断前面有多少个偶数，如果为偶数判断前面有多少个奇数。

AC代码

class Solution {
public:
    typedef long long ll;
    ll mod=1000000007;
    int numOfSubarrays(vector<int>& arr) {
        int ans0=1,ans1=0;
        int sum=0;
        ll res=0;
        for(int i=0;i<arr.size();i++)
        {
            sum+=arr[i];
            sum%=2;
            if(sum==1)
            {
                res+=ans0;
                ans1++;
            }
            else
            {
                res+=ans1;
                ans0++;
            }
            res%=mod;
        }
        return int(res);
    }
};