描述
给你一个整数数组 nums 和一个整数 k。如果某个连续子数组中恰好有 k 个奇数数字,我们就认为这个子数组是「优美子数组」。
请返回这个数组中 「优美子数组」 的数目。
分析
原理是计算k个奇数区间两端的偶数个数,然后相乘就是这个k个奇数区间的子数组个数。
滑动窗口遍历数组。右指针不停遍历直到奇数的个数等于k。
奇数的个数等于k,标记right的位置,然后仍然向右遍历直到走到末尾或者遇到下一个奇数,计算这段区间的个数。
class Solution {
public int numberOfSubarrays(int[] nums, int k) {
int right = 0, left = 0, ans = 0, count = 0;
while (right < nums.length) {
if ((nums[right] & 1) == 1) {
count++;
}
right++;
if (count == k) {
int tmpr = right;
while (right < nums.length && (nums[right] & 1) == 0) {
right++;
}
int r = right - tmpr + 1;
int tmpl = left;
while ((nums[left] & 1) == 0) {
left++;
}
int l = left - tmpl + 1;
ans += l * r;
left++;
count--;
}
}
return ans;
}
}