统计「优美子数组」

统计「优美子数组」

题目描述

给你一个整数数组 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 <= nums.length <= 50000
1 <= nums[i] <= 10^5
1 <= k <= nums.length

解题思路

样例分析

[2,2,2,1,2,2,1,2,2], k = 2 答案为12
那么答案是怎么来的呢？

首先统计每个奇数前后各有多少个偶数：
在这个样例中有两个奇数，[1,1]
用两个数组，分别储存每个奇数前面有多少个偶数，以及后面有多少个偶数
每个奇数前面的偶数数量[3,2]
每个奇数后面的偶数数量[2,2]
注意：统计前面偶数数量的时候，到前面一个奇数为止，例如第二个1,前面（到第一个1）就两个偶数而不是5个，统计后面的偶数也是同样

然后统计每段k个相邻的奇数可以有多少种优美子数组
k = 2 正好有两个奇数，而第一个奇数前面有三个偶数，说明优美子数组的最开始可以有0,1,2,3个偶数；第二个奇数后面有2个偶数，说明优美子数组的结尾可以有0,1,2个偶数，则结果就是(3+1)*(2+1)=12

到这里这个题的基本思路就结束了，然后再看一个长一点的示例：
[2,1,2,2,1,2,2,2,1,2,2,2,2,1,2,2,2,1,2,2,1,2,2,2], k = 3
每个奇数前面的偶数数量[1,2,3,4,3,2]
每个奇数后面的偶数数量[2,3,4,3,2,3]
k = 3则:

第一段3个相邻奇数为第1-3个奇数，然后这一部分的结果为(1+1)(4+1)=10
第二段3个相邻奇数为第2-4个奇数，然后这一部分的结果为(2+1)(3+1)=12
第二段3个相邻奇数为第3-5个奇数，然后这一部分的结果为(3+1)(2+1)=12
第二段3个相邻奇数为第4-6个奇数，然后这一部分的结果为(4+1)(3+1)=20
最终答案10+12+12+20=54

相关代码

int numberOfSubarrays(int* nums, int numsSize, int k){
    int a[50000],b[50000],i,j,x,t=0,sum=0,q=0;
    for(i=0,j=0,x=0;i<numsSize;i++){//a数组储存每个奇数前偶数的个数；b数组储存每个奇数后偶数的个数； 
        if(nums[i]%2){
            q++;
        	a[j++]=t;
            if(j>1){
                b[x++]=t;
            }
            t=0;
        }
        else{
            t++;
        }
    }
    b[x++]=t;
    for(i=0;i<=x-k;i++){
        sum+=(a[i]+1)*(b[i+k-1]+1);
    } 
    if(q)//判断nums数组中是否含有奇数；
    return sum;
    return 0;
}