1. 题目描述
给你一个整数数组 nums 和一个整数 k。
如果某个 连续 子数组中恰好有 k 个奇数数字,我们就认为这个子数组是「优美子数组」。
请返回这个数组中「优美子数组」的数目。
示例 1:
输入:nums = [1,1,2,1,1], k = 3
输出:2
解释:包含 3 个奇数的子数组是 [1,1,2,1] 和 [1,2,1,1] 。
示例 2:
输入:nums = [2,4,6], k = 1
输出:0
解释:数列中不包含任何奇数,所以不存在优美子数组。
示例 3:
输入:nums = [2,2,2,1,2,2,1,2,2,2], k = 2
输出:16
提示:
1 <= nums.length <= 50000
1 <= nums[i] <= 10^5
1 <= k <= nums.length
2. 题目解析
其实这个题目的解法跟上一篇博客介绍的方法也差不多,主要是前缀和加上差分的思想。
考虑以 i 结尾的「优美子数组」个数,我们需要统计符合条件的下标 j 的个数,其中0≤j≤i 且 [j..i]这个子数组里的奇数个数恰好为 k 。如果枚举[0..i] 里所有的下标来判断是否符合条件,那么复杂度将会达到 O(n^2)无法通过所有测试用例,因此我们需要优化枚举的时间复杂度。
我们定义pre[i]为[0..i] 中奇数的个数,则 pre[i] 可以pre[i−1] 递推而来,即:
那么[j..i] 这个子数组里的奇数个数恰好为 k 这个条件我们可以转化为:
简单移项可得符合条件的下标 j需要满足:
所以我们考虑以 i结尾的「优美子数组」个数时只要统计有多少个奇数个数为pre[i]−k 的pre[j] 即可。我们只要建立频次数