leetcode 1248. 统计「优美子数组」

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

我的代码

class Solution {
public:
    int len;
    int res;
    vector<bool> flag;


    void findSubArray(int loc, int k) {
        if (k == 0) {
            ++res;
        }
        if ((k < 0) || (loc >= len)) {
            return;
        }
        if (!flag[loc]) {
            findSubArray(loc + 1, k);
        } else {
            findSubArray(loc + 1, k - 1);
        }
    }

    int numberOfSubarrays(vector<int>& nums, int k) {
        len = nums.size();
        if (len < k) {
            return 0;
        }
        for (auto num: nums) {
            flag.emplace_back(num % 2 == 1);
        }
        res = 0;
        for (int i = 0; i + k <= len; ++i) {
            if (!flag[i]) {
                findSubArray(i + 1, k);
            } else {
                findSubArray(i + 1, k - 1);
            }
        }
        return res;
    }
};

我的成绩

执行结果：超出时间限制

一些想法

本道题我的方法是暴力法，最后超出了时间限制。看完题解我觉得没做出来很正常，实在是想不出这两种方法。