题面
给你一个整数数组 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
题解
我的题解
遍历并得到所有奇数在原向量中的下标存入新向量，从第一个奇数开始，每次找k个奇数，并以下标向两侧扩展，得到左右端点可能的组合，进而得到结果。
class Solution
{
public:
int numberOfSubarrays(vector<int>& nums, int k)
{
if (k > nums.size())
{
return 0;
}
int ans = 0;
vector<int> odd;
for (int i = 0; i < nums.size(); ++i)
{
if (nums[i] % 2 == 1)
{
odd.push_back(i);
}
}
for (int i = 0; i + k - 1 < odd.size(); ++i)
{
int left = 0, right = 0;
if (i == 0)
{
left = odd[0] - 0 + 1;
//right = odd[0 + k] - odd[0 + k - 1];
}
else
{
left = odd[i] - odd[i - 1];
}
if (i + k == odd.size())
{
//left = odd[i] - odd[i - 1];
right = (nums.size() - 1) - odd[i + k - 1] + 1;
}
else
{
right = odd[i + k] - odd[i + k - 1];
}
ans += left * right;
}
return ans;
}
};
“前缀和”
class Solution {
vector<int> cnt;
public:
int numberOfSubarrays(vector<int>& nums, int k) {
int n = (int)nums.size();
cnt.resize(n + 1, 0);
int odd = 0, ans = 0;
cnt[0] = 1;
for (int i = 0; i < n; ++i) {
odd += nums[i] & 1;
ans += odd >= k ? cnt[odd - k] : 0;
cnt[odd] += 1;
}
return ans;
}
};
