题意:连续的子序列中有k个奇数的有多少组,个数不限。
arr数组用来记录奇数在原数组中的下标。
例如[2,2,2,1,2,2,1,2,2,2],符合要求的最小子序列是[1,2,2,1],这个序列可以向前、后扩展,向前(四种情况)可以不扩展、扩展1位、两位、三位;向后(四种情况)可以不扩展、扩展1位、两位、三位,它只有一个最小子序列,所以结果是44=16。
arr数组记录奇数下标,相减就可以得到扩展的情况个数,arr=[-1,3,6,10],k=2,所以(arr[i]-arr[i-1])(arr[i+k]arr[i+k-1]),i=1时即(arr[1]-arr[0])(arr[3]-arr[2])。
C++
class Solution {
public:
int numberOfSubarrays(vector<int>& nums, int k) {
int len=nums.size();
int arr[len+2];//这俩存边界
int j=0;
int res=0;
for(int i=0;i<len;i++){
if(nums[i]%2!=0){
arr[++j]=i;
}
}
arr[0]=-1;
arr[j+1]=len;
for(int i=1;i+k<j+2;i++){
res+=(arr[i]-arr[i-1])*(arr[i+k]-arr[i+k-1]);
}
return res;
}
};