1248. 统计「优美子数组」（java）

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

提示：

1 <= nums.length <= 50000
1 <= nums[i] <= 10^5
1 <= k <= nums.length

---

解法一：滑动窗口

• 步骤
  • 找出k个奇数的左右边界
  • 找出首个奇数前面数字边界及第k个奇数的下一个奇数边界，计算个数
  • (leftEvenCnt + 1) * (rightEvenCnt + 1)
• 参数说明
  • left 、right为当前k个奇数的最小连续数组范围
  • leftcnt、rightcnt为除开k个奇数与前后一个奇数的个数（可以进行组合成优美子数组）

class Solution {
    public int numberOfSubarrays(int[] nums, int k) {
        int left=0,right=0,oddCnt=0,res=0;
        while(right<nums.length){
            //或者用nums[right++]&1
            //右指针先走
            if((nums[right++]%2)==1){
                oddCnt++;
            }
            //有k个奇数了
            if(oddCnt==k){
                //开始计算到右边界个数
                int temp=right;
                while(right<nums.length&&(nums[right]%2)==0) {
                    right++;
                }
                int rightcnt=right-temp;
                //计算到左边界个数
                int leftcnt=0;
                while((nums[left]%2)==0){
                    left++;
                    leftcnt++;
                }
                //当前可组合的数组数
                res+=(rightcnt+1)*(leftcnt+1);
                //去掉当前数组的首个奇数
                left++;
                oddCnt--;
            }
        }
        return res;
    }
}

---

解法二：前缀和

• 用一个数组prefixCnt存储
  • 下标是[奇数和]，值是[奇数和的个数]
  • sum-k就是要累加的个数，如sum=2 k=2时 在蓝色部分情况共4次，每次遇见该情况都计算前缀和，即黄色数3+1次，
  • 

class Solution {
    public int numberOfSubarrays(int[] nums, int k) {
        int[] prefixCont = new int[nums.length+1];
        prefixCont[0]++;
        int sum=0,res=0;
        for(int num:nums){
            sum+=num%2;
            prefixCont[sum]++;
            if(sum>=k){
                res+=prefixCont[sum-k];
            }
        }
        return res;
    }
}