Leetcode1248. 统计「优美子数组」——python求解

统计「优美子数组」

题目：

给你一个整数数组 nums 和一个整数 k。

如果某个 连续 子数组中恰好有 k 个奇数数字，我们就认为这个子数组是「优美子数组」。

请返回这个数组中「优美子数组」的数目。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-number-of-nice-subarrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例 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

下述方法是暴力求解，但是可惜超出时间限制了

class Solution:
    def numberOfSubarrays(self, nums: List[int], k: int) -> int:
        n = len(nums)
        ans = 0
        for i in range(n):
            j = i
            flag = 0
            while flag <= k and j < n:
                if nums[j] % 2 == 1:
                    flag += 1
                j += 1
                if flag == k:
                    ans += 1
            if flag < k:
                break
        return ans

所以下述方法是通过例子得出的通解

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