给你一个整数数组 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[i]即前i个数的和,当前sum[i]%2= =0说明是偶数,那么我们找前i个数中有多少个sum[j]%2= =1的个数与sum[i]搭配就能构造出(sum[i]-sum[j])是奇数,同理当前sum[i]%2= =1说明是奇数,那么去找偶数即可。另外也要统计sum[i]%2= =1本身,例如1 3 5,统计sum[1]=1,我们要加上偶数个数,这个时候偶数even=0,我们需初始化even=1,就能加上本身sum[i]%2==1这个了。
class Solution {
public:
int numOfSubarrays(vector<int>& arr) {
int odd=0,even=1,sum=0,res=0;
for(int i=0;i<arr.size();i++){
sum+=arr[i];
if(sum&1){
res+=even;
res=res%(1000000007);
odd++;
}else{
res+=odd;
res=res%(1000000007);
even++;
}
}
return res;
}
};