leetcode1524. 和为奇数的子数组数目（Python3、Cpp）

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

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

---

方法：前缀和+动态规划

思路：

本题是一道典型的前缀和的问题。而每次求子数组的和是不是奇数的时候，需要用到前面数组的前缀和，因此可以写成动态规划的形式之后再优化。

我们定义二维dp数组，dp(i,0)表示以arr[i]为结尾的子数组中，和为奇数的数组个数；dp(i,1)表示以arr[i]为结尾的子数组中，和为偶数的数组个数。

那么，我们的答案应该就是sum(dp[i][0])。

下面考虑状态转移方程，对于dp(i)来说，只需要考虑dp(i-1)即可，

• 如果此时arr[i]为奇数，那么偶数+奇数=奇数，奇数+奇数=偶数，则：
  $$\begin{gathered} dp[i][0]=dp[i-1][1]+1 \\ dp[i][1]= \end{gathered}$$