和为奇数的子数组数目

1524. 和为奇数的子数组数目

难度中等18收藏分享切换为英文接收动态反馈

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

动态规划

注意初始条件：odd=0，even=1；因为空的前缀的和是 0，也是偶数前缀和。

class Solution {
    // 动态规划
    // 如果当前的和为奇数,那么需要加上和为偶数时的数量
    // 如果当前的和为偶数,那么需要加上和为奇数时的数量
    public int numOfSubarrays(int[] arr) {
        int MOD=(int)1e9+7;
        int sum=0;
        int n=arr.length;
        // 分别使用 odd 和 even 表示奇数前缀和的数量与偶数前缀和的数量。
        // 初始时，odd=0，even=1，因为空的前缀的和是 0，也是偶数前缀和。
        int odd=0,even=1;
        int res=0;

        for(int i=0;i<n;i++){
            sum+=arr[i];
            if(sum%2==0){
                res=(res+odd)%MOD;
                even++;
            }else{
                res=(res+even)%MOD;
                odd++;
            }
        }
        return res;
    }
}

class Solution {
    int mod=(int)1e9+7;
    public int numOfSubarrays(int[] arr) {
        int n=arr.length;
        int sum=0;
        int odd=0,even=1;
        int ans=0;

        for(int i=0;i<n;i++){
            sum+=arr[i];
            // 如果当前前缀和是奇数，那么他加所有之前偶数的数量。 
            // 如果当前前缀和是偶数，那它只能加奇数的数量。 
            if(sum%2==0){
                ans=(ans+odd)%mod;
                even++;
            }else{
                ans=(ans+even)%mod;
                odd++;
            }
        }
        return ans;
    }
}

赏

谢谢你请我吃糖果

微信