题目描述:
给你一个整数数组 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:
由于我们已经记录了每个奇数的下标,所以我们知道对于第 i i i 个奇数,它的前一个奇数的下标为 o d d [ i − 1 ] odd[i−1] odd[i−1],也就是说 ( o d d [ i − 1 ] , o d d [ i ] ) (odd[i−1],odd[i]) (odd[i−1],odd[i]) 间的数都为偶数。同理可得 ( o d d [ i + k − 1 ] , o d d [ i + k ] ) (odd[i+k−1],odd[i+k]) (odd[i+k−1],odd[i+k])间的数也都为偶数。那么我们可以得出满足 l ∈ ( o d d [ i − 1 ] , o d d [ i ] ] l∈(odd[i−1],odd[i]] l∈(odd[i−1],odd[i]] 且 r ∈ [ o d d [ i + k − 1 ] , o d d [ i + k ] ) r∈[odd[i+k−1], odd[i+k]) r∈[odd[i+k−1],odd[i+k]) 条件的子数组 [ l , r ] [l,r] [l,r] 包含 [ [ o d d [ i ] , o d d [ i + k − 1 ] ] [[odd[i],odd[i+k−1]] [[odd[i],odd[i+k−1]] 且 [ l , r ] [l,r] [l,r] 里的奇数个数为 k k k 个。因此对于第 i i i个奇数,它对答案的贡献为符合条件的 [ l , r ] [l,r] [l,r] 的个数,即:
( o d d [ i ] − o d d [ i − 1 ] ) ∗ ( o d d [ i + k ] − o d d [ i + k − 1 ] ) (odd[i]−odd[i−1])∗(odd[i+k]−odd[i+k−1]) (odd[i]−odd[i−1])∗(odd[i+k]−odd[i+k−1])
我们只要遍历一遍 o d d odd odd 数组即可求得最后的答案,注意边界的处理。
代码1:
class Solution:
def numberOfSubarrays(self, nums, k):
n = len(nums)
odd = [-1]
ans = 0
for i in range(n):
if nums[i] % 2 == 1:
odd.append(i)
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
s = Solution()
nums = [1, 1, 2, 1, 1]
k = 3
print(s.numberOfSubarrays(nums, k))
代码2:
class Solution:
def numberOfSubarrays(self, nums, k):
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
题目来源:
https://leetcode-cn.com/problems/count-number-of-nice-subarrays