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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路

• 遍历数组记录第i个奇数，在数组中的下标位置，保存在vOdd中。
• 利用滑动窗口的思想，保持一个含有k个奇数的“窗口”，即vOdd[i]到vOdd[i+k-1]之间的k个奇数。
• 设上述窗口中的左指针为l，右指针为r，则l的取值范围为(vOdd[i-1], vOdd[i] ]；r的取值范围为[ vOdd[i+k-1], vOdd[i+k] )。[l,r]所对应下标之间的子数组即为优美子数组。left = vOdd[i] - vOdd[i-1]代表第i-1个奇数到第i个奇数之间偶数的个数，同理right = vOdd[i+k] - vOdd[i+k-1]代表第i+k-1个奇数和第i+k个奇数之间偶数的个数。由此可以计算出，包含这k个奇数的子数组个数为left * right。
• 遍历vOdd，计算以第i个奇数为起点的k个奇数，所能得到的优美子数组的累加和。
• 注意边界情况的处理，详细请见代码。

三、解题代码

class Solution {
public:
    int numberOfSubarrays(vector<int>& nums, int k) {
        int n = nums.size();
        int cnt = 0;
        vector<int> vOdd(n+2, -1);
        for(int i = 0; i < n; ++i)
        {
            if((nums[i] & 1))
                vOdd[++cnt] = i; //记录第几个奇数所对应的下标位置
        }
        vOdd[++cnt] = n;//  为了统计最后一个奇数到数组结尾之间偶数的个数
        int res = 0;
        for(int i = 1; i + k <= cnt; ++i)
        {
            res += (vOdd[i] - vOdd[i-1])*(vOdd[i+k] - vOdd[i+k-1]);
        }
        return res;
    }
};

四、解题结果