滑动窗口 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 <= 50000
1 <= nums[i] <= 10^5
1 <= k <= nums.length

解题思路

先遍历数组将所有的奇数的下标找出来，再使用滑动窗口来计算，滑动窗口的长度即为k。记录了每个奇数的下标，所以我们知道对于第 ii个奇数，它的前一个奇数的下标为queue[i-1]，也就是说 queue[i-1] ~ queue[i] 间的数都为偶数。同理可得 queue[i+k-1] ~ queue[i+k] 间的数也都为偶数。所有第i个奇数贡献的个数为(queue[i] ~ queue[i-1])*(queue[i+k] ~ queue[i+k-1])

本题主要关注边界问题。

在队列的首位添加两个哨兵可以解决边界问题，队首添加-1，队尾添加找出来的奇数的数量n。

class Solution:
    def numberOfSubarrays(self, nums: List[int], k: int) -> int:
        # 先遍历数组将所有的奇数的下标找出来，再使用滑动窗口来计算
        # 在队列的首位添加两个哨兵可以解决边界问题
        ret = 0
        queue = [-1]
        length = len(nums)
        for i in range(length):
            if nums[i]%2==1:
                queue.append(i)
        queue.append(length)

        for i in range(1, len(queue) - k):
            ret += ((queue[i] - queue[i - 1])*(queue[i + k] - queue[i + k - 1]))

        return ret