字首和-5457. 和為奇數的子陣列數目
阿新 • • 發佈:2020-07-26
2020-07-2619:20:40
問題描述:
給你一個整數陣列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 。
輸入: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
問題求解:
採用字首和的思路即可,計數當前的presum和之前的奇偶presum數量,就可以得到最終的解。
本題的一個提示是”子陣列“。
int mod = (int)1e9 + 7; public int numOfSubarrays(int[] arr) { int res = 0; int presum = 0; int odd = 0; int even = 1; for (int num : arr) { presum += num; if (presum % 2 == 0) { res = (res + odd) % mod; even += 1; } else { res = (res + even) % mod; odd += 1; } } return res; }