1524. 和为奇数的子数组数目(C语言简单易懂)

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

解析:这道题考察的是动态规划+数组

我们首先要知道奇数+奇数=偶数,奇数+偶数=奇数。我们可以用位运数符进行判断,(arr [i ] & 1 ) == 1,如果为true,那么为奇数,否则为false。

int numOfSubarrays(int* arr, int n){
    int MOD=(int) 1e9+7;
    int odd=0,even=0,cnt=0;
        for(int i=0;i<n;i++){
              // odd 表示以当前元素 k 结尾和为奇数子数组数目
            // even 表示以当前元素 k 结尾和为偶数子数组数目
            if((arr[i]& 1) == 1){
                int t = odd;
                // 偶数 + 奇数k = 奇数              
                odd = even + 1; // k 为奇数,  {k, 偶数1+k, 偶数2 + k, ...}
                even = t;       // 奇数 + 奇数k = 偶数
            }else{
                even++;  // k 为偶数,  {k, 偶数1+k, 偶数2 + k, ...}
            }

            cnt = (cnt + odd) % MOD; 
    }
  return cnt;
}

猜你喜欢

转载自blog.csdn.net/zhi6fui/article/details/124261441