LeetCode-1248、统计「优美子数组」-中等

最新推荐文章于 2022-06-04 17:30:24 发布

clover_my

最新推荐文章于 2022-06-04 17:30:24 发布

阅读量148

点赞数

分类专栏： LeetCode题库

本文链接：https://blog.csdn.net/clover_my/article/details/105666984

版权

LeetCode题库专栏收录该内容

218 篇文章 6 订阅

订阅专栏

LeetCode-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

代码：

class Solution:
    def numberOfSubarrays(self, nums: List[int], k: int) -> int:
        ans = 0
        count = 1
        evennum = []
        for n in nums:
            if n%2 == 1:
                evennum.append(count)
                count = 1
            else:
                count += 1
        else:
            evennum.append(count)
        for i in range(k, len(evennum)):
            ans += evennum[i] * evennum[i-k]
        return ans