整数配列 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;
}