leetcode-1248. 统计「优美子数组」

 

---

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^