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;
}
};
我的成绩
执行结果:超出时间限制
一些想法
本道题我的方法是暴力法,最后超出了时间限制。看完题解我觉得没做出来很正常,实在是想不出这两种方法。