Leetcode 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-number-of-nice-subarrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

class Solution {
public:
    int numberOfSubarrays(vector<int>& nums, int k) {
        int ans = 0,num = 0,len = nums.size();
        int i = 0,tmp = 0;
        vector<int> pre(len),post(len);
        vector<int> idxs;
        for(i=0;i<len;++i){
            if(nums[i]%2==0) pre[i] = ++num;
            else{
                idxs.push_back(i);
                pre[i] = num;
                num = 0;
            }
        }
        i=0;num=0;
        for(i=len-1;i>=0;--i){
            if(nums[i]%2==0) post[i] = ++num;
            else{
                post[i] = num;
                num = 0;
            }
        }
        i = 0;len = idxs.size();
        while(i<len){
            int j = tmp;
            while(j<len && j-i+1<k) ++j;
            if(j>=len) break;
            ans += (pre[idxs[i]]+1)*(post[idxs[j]]+1);
            tmp = j;
            ++i;
        }
        return ans;
    }
};