每日一题-1248. 统计「优美子数组」

今天是2020年4月21日，星期二。

题目描述

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

题目分析

今日这道题还是参考了甜姨的题解：统计「优美子数组」，可以帮甜姨点赞转发。

第一种解法是借助于「滑动窗口」的思想，「滑动窗口」的思想进一步简化为了双指针问题，使用$left$和$right$两个指针用来限定子数组的左右边界。当前$right$所指的位置不为奇数时，则继续向后移动；否则区间奇数个数加1。当区间奇数个数 == k时，计算此时这种情况的「优美子数组」的个数。

在计算「优美子数组」的个数时，奇数的个数一旦确定，就由偶数的个数决定了数组的个数；所以寻找区间内第一个奇数左边的偶数个数$leftEventCount$和第k个奇数右面的偶数个数$rightEventCount$(注意：第k个奇数右面的偶数的个数的计算方法，到达数组的边缘或者遇到下一个奇数)。由于组成的子数组中可以选择$0—leftEventCount/rightEventCount$之间的任何一个整数，所以子数组的个数就是$(leftEventCount + 1) * (rightEventCount + 1)$。

第二中解法是利用了前缀和的方式，这种方式小编还不能很好的掌握，靠大家细细体会了。有掌握了的小伙伴可以私信小编，交流一下。

参考代码

• 滑动窗口参考代码

/**
 * 双指针，滑动窗口解法 @Sweetiee 甜姨的题解
 * 不断扩大right的值，直到区间中含有k个奇数
 *
 * @param nums
 * @param k
 * @return
 */
public int numberOfSubarrays(int[] nums, int k) {
    int length = nums.length;

    if (length <= 0) {
        return 0;
    }
    // 左右指针
    int left = 0;
    int right = 0;
    // 奇数的个数
    int oddCount = 0;
    int result = 0;

    while (right < length) {
        // 使用位运算代替取余
        if ((nums[right++] & 1) == 1) {
            oddCount++;
        }

        // 说明当前的区间中已经含有了k个奇数，来确定优美子数组的个数
        if (oddCount == k) {
            // 向将区间的右边界继续向右扩展，直到出界或者遇到下一个奇数
            // 此时这个区间中的第k个元素的右边的偶数就计算出来了
            int tmp = right;
            while (right < length && (nums[right] & 1) == 0) {
                right++;
            }
            // 此时退出循环，即right出界或者遇到下一个奇数，之间的就是偶数
            int rightEventCount = right - tmp;
            int leftEventCount = 0;
            // 找到区间内的第一个奇数，则在第一个奇数之前的均为区间内的整数，均可以选为 优美子数组
            while ((nums[left] & 1) == 0) {
                leftEventCount++;
                left++;
            }

            // 第 1 个奇数左边的 leftEventCount 个偶数都可以作为优美子数组的起点
            // (因为第1个奇数左边可以1个偶数都不取，所以起点的选择有 leftEvenCnt + 1 种）
            // 第 k 个奇数右边的 rightEventCount 个偶数都可以作为优美子数组的终点
            // (因为第k个奇数右边可以1个偶数都不取，所以终点的选择有 rightEvenCnt + 1 种）
            // 所以该滑动窗口中，优美子数组左右起点的选择组合数为 (leftEventCount + 1) * (rightEventCount + 1)
            result += (leftEventCount + 1) * (rightEventCount + 1);

            // 该区间统计结束后，继续到下一个区间，此时left应该指向下一个元素。
            left++;
            oddCount--;
        }
    }

    return result;
}

• 前缀和参考代码

/**
 * 前缀和的方式，通过debug方式查看了执行情况。还是有点不理解
 *
 * @param nums
 * @param k
 * @return
 */
public int numberOfSubarrays2(int[] nums, int k) {
    // prefixCount的下标是前缀和（即当前奇数的个数），值是前缀和的个数
    int[] prefixCount = new int[nums.length + 1];
    // 下标0，表示当前为0个奇数；值为1，则表明当前有1个下标
    prefixCount[0] = 1;

    // 遍历原数组，计算当前的前缀和，统计到 prefixCnt 数组中，
    // 并且在 res 中累加上与当前前缀和差值为 k 的前缀和的个数。
    int sum = 0;
    int result = 0;
    for (int num : nums) {
        // 如果num为奇数，则sum的值加一；否则sum维持原值
        sum += (num & 1);
        prefixCount[sum]++;
        if (sum >= k) {
            result += prefixCount[sum - k];
        }
    }

    return result;
}