统计「优美子数组」

给你一个整数数组 nums 和一个整数 k。

如果某个连续子数组中恰好有 k 个奇数数字，我们就认为这个子数组是「优美子数组」。

请返回这个数组中「优美子数组」的数目。

在这道题目中，我首先想到的方法是将奇数数组的下标保留下来，之后根据乘法原理计算有多少个优美子数组即可，我的程序代码如下：

#include <iostream>
#include <vector>
using namespace std;

int numberOfSubarrays(vector<int>& nums, int k);
int main()
{
    int num = 0;
    int k = 0;
    cin >> num >> k;
    vector<int> nums;
    for (int i = 0; i < num; ++i)
    {
        int t;
        cin >> t;
        nums.push_back(t);
    }
    cout << numberOfSubarrays(nums, k);
    return 0;
}

int numberOfSubarrays(vector<int>& nums, int k) {
    vector<int> arr;
    int ans = 0;
    for (int i = 0; i < nums.size(); ++i)
    {
        if (nums[i] % 2)
        {
            arr.push_back(i);
        }
    }
    if (k > arr.size())
    {
        return 0;
    }
    for (int i = 0; i + k - 1 < arr.size(); ++i)
    {
        int l = 0, r = 0;
        if (i == 0)
        {
            l = arr[i] + 1;
        }
        else
        {
            l = arr[i] - arr[i - 1];
        }
        if (i + k - 1 == arr.size() - 1)
        {
            r = nums.size() - arr[i + k - 1];
        }
        else
        {
            r = arr[i + k - 1] - arr[i + k];
        }
        ans += (l * r);
    }
    return ans;
}

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