1248. 统计「优美子数组」-统计奇数下标

1248. 统计「优美子数组」

给你一个整数数组 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

这一题的求解我们应该是先统计奇数的下标，这样就可以计算满足k和奇数的子数组数量了，解题代码如下：

int numberOfSubarrays(int* nums, int numsSize, int k){
    int count=0;
    int r[numsSize];
    for(int i=0;i<numsSize;i++){
        if(nums[i]%2!=0){
           r[count++]=i;
        }
    }
    if(count<k){
        return 0;
    }
    int sum=0;

    
    for(int i=0;i<=count-k;i++){
        if(i==count-k){ 
            
            if(i!=0){

            
            sum=sum+(numsSize-r[i+k-1])*(r[i]-r[i-1]);
            }
            else{
                     sum=sum+(numsSize-r[i+k-1])*(r[i]+1);

            }
            break;
        }

        if(i==0){
            sum=sum+(r[i]+1)*(r[i+k]-r[i+k-1]);
        }
        else{
            sum=sum+(r[i+k]-r[i+k-1])*(r[i]-r[i-1]);
        }
      
    }
    return sum;

}