class Solution {
public:
int numberOfSubarrays(vector<int>& nums, int k) {
}
};
1248. 统计「优美子数组」
给你一个整数数组 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 <= 500001 <= nums[i] <= 10^51 <= k <= nums.length原站题解
python3 解法, 执行用时: 704 ms, 内存消耗: 22.2 MB, 提交时间: 2022-08-30 11:51:02
class Solution:
def numberOfSubarrays(self, nums: List[int], k: int) -> int:
cnt = [0] * (len(nums) + 1)
cnt[0] = 1
odd, ans = 0, 0
for num in nums:
if num % 2 == 1:
odd += 1
if odd >= k:
ans += cnt[odd - k]
cnt[odd] += 1
return ans
python3 解法, 执行用时: 720 ms, 内存消耗: 22.2 MB, 提交时间: 2022-08-30 11:50:33
class Solution:
'''
统计所有奇数，然后根据公式
'''
def numberOfSubarrays(self, nums: List[int], k: int) -> int:
n = len(nums)
odd = [-1]
ans = 0
for i in range(n):
if nums[i] % 2 == 1:
odd.append(i)
odd.append(n)
print(odd)
for i in range(1, len(odd) - k):
ans += (odd[i] - odd[i - 1]) * (odd[i + k] - odd[i + k - 1])
return ans