leetcode - 1248. 统计「优美子数组」

原题链接：https://leetcode-cn.com/problems/count-number-of-nice-subarrays/

题目：

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

题解：

首先我们把数组映射成一个0 1数组，0表示偶数，1表示奇数

例如：[0, 0, 0, 1, 0, 0, 1 ,0 , 0, 0, 1, 0, 0, 1]  k = 2

寻找过程：

1. 先找到前k 个奇数 [0, 0, 0, 1, 0, 0, 1 ,0 , 0, 0, 1, 0, 0, 1]

2 . 判断第一个奇数的前面有几个连续偶数，判断第k个奇数后面有几个连续的偶数 [0, 0, 0, 1, 0, 0, 1 ,0 , 0, 0, 1, 0, 0, 1] 

3. 那么对于这个k个奇数，符合的区间个数为4 * 4个

4. 寻找下k个奇数[0, 0, 0, 1, 0, 0, 1 ,0 , 0, 0, 1 , 0, 0, 1]

5. 执行步骤2，[0, 0, 0, 1, 0, 0,1 ,0 , 0, 0, 1 ,0, 0,1]

6. 对于这k个奇数，符合的区间个数为 3 * 3

7. 以此类推...

8. 考虑几个边界情况

      （1）最后一个值为奇数且满足k个奇数的情况

      （2）总的奇数小于k个

9. 我的代码实现中没有在原来的数组上操作，而是将奇数的位置copy到单独的一个数组中进行计算

int numberOfSubarrays(int* nums, int numsSize, int k){
    int add[numsSize + 3];
    int top = 0;
    add[top ++] = -1;
    for(int i = 0; i < numsSize;i++) {
        if(nums[i]%2==1) add[top++] = i;
    }
    if(top <= k) return 0;

    int mx = 0;
    for(int i = k;i<top;i++) {
        int l = i - k + 1;
        int x = add[l] - add[l-1];
        int y = 1;
        if (i + 1 < top) {
            y = add[i+1] - add[i];
        } else {
            y = numsSize - add[i];
        }
        mx += x * y;
    }

    return mx;
}