前缀和算法应用——周赛题

5457. 和为奇数的子数组数目

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

提示：

• 1 <= arr.length <= 10^5
• 1 <= arr[i] <= 100

思路：如果当前的和为1，那么找到前缀和为0的，在这一段中，和就为奇数因为前缀和为0的变成前缀和为1的，中间的和肯定是一个奇数同理，如果当前的和为0，那么找到前缀和为1的，在这一段中，和就为奇数我们只需要统计前缀和为0和1的个数即可

	public int numOfSubarrays(int[] arr) {
        int[] counts = new int[2];
        int num = 0;
        int ans = 0;
        counts[0] = 1;
        int mod = 1000000007;
        for(int x : arr){
            num = (num+x)%2;
            ans += counts[1-num];
            counts[num]++;
            ans %= mod;
        }
        return ans%mod;
    }