给你一个整数数组 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
假设原数组有每m个奇数,将每一个奇数的位置保存到store数组中,第i个奇数位置就是store[i],由于题目要求是连续数组,所以每个符合题意的子数组其中必定包含着连续的k(k是输入)个奇数,只要确定了中心的这k个奇数,从中间向两边延伸就能得到全部的子数组了。
向两边延伸的边界就是又碰上了一个奇数或者到达了原数组的边界,详见代码。
class Solution {
public int numberOfSubarrays(int[] nums, int k) {
int len = nums.length;
int store[] = new int [len+2];
int odd = 0;
for(int i=0;i<len;i++)
{
if(nums[i]%2!=0)
store[++odd] = i;
}
store[0] = -1;
store[++odd] = len;
int ans = 0;
for(int i=1;i+k-1<odd;i++)
{
int l,r;
l = store[i] - store[i-1] - 1;
r = store[i+k] - store[i+k-1] - 1;
ans = ans + ((l+1)*(r+1));
}
return ans;
}
}