leetcode 第 31 场双周赛和为奇数的子数组数目

和为奇数的子数组数目




给你一个整数数组 arr 。请你返回和为 奇数 的子数组数目。
由于答案可能会很大,请你将结果对 10^9 + 7 取余后返回。




题解
首先明白一点:


奇数+偶数=奇数
偶数+奇数==奇数


那么问题就可以转化成动态规划问题,我们遍历数组的每一个数,如果当前的是奇数,那么我们就要找到多之前有多少个偶数的前缀和,当前如果是偶数,那么就需要找到之前有多少个奇数的前缀和,原因是之前有多少种奇数数组组合,当出现了一个新的能组成奇数的,那么它可以和之前的每一个组合组成一个奇数的子数组。

public class numOfSubarrays {

    public static int numOfSubarrays(int[] arr){
       int ans=0;
       //记录前缀和是奇数的次数
       int odd=0;
       //记录前缀和是偶数的次数
       int even=1;
       int sum=0;
       for (int num:arr){
           sum=(sum+num)%2;
           if ((sum&1)==1){
               ans+=even;
               odd++;
           }else {
               ans+=odd;
               even++;
           }
        ans%=1000000007;
       }
       return ans;
    }
}

你可能感兴趣的:(动态规划,动态规划,leetcode,算法,数据结构)