给你一个整数数组 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
思路:
这道题目可以等价变换为:奇数视为1,偶数视为0,统计有多少个子数组和为k(包含了k个1)
那么我们容易联想到前缀和的概念。
设给出的序列的前n项中存在S[n]个奇数,那么以a[i]为起点,a[j]为终点的子序列中,存在S[j] - S[i-1] 个奇数。
我们只需找到一个容器,来记录存在多少个这样的i,j能符合S[j]-S[i-1]=k。我用的是unordered_map memo。
代码中preSum动态地表示在[0,j-1]以内的奇数项数S[j]。
每次在j后移时,我需要确切地知道有多少个i能使得S[j]-S[i-1]=k,显然S[i-1]=S[j]-k,因此嵌套进memo即可。res += memo[preSum-k]。
class Solution {
public:
int numberOfSubarrays(vector<int>& nums, int k)
{
unordered_map<int, int> memo;//HashMap:假设val是数组元素i前缀和的值,则存在memo[val]个前缀和为val的i
memo[0] = 1;
int result = 0;
int preSum = 0;
for (auto& it : nums)
{
preSum += it & 1;
memo[preSum]++;
if (preSum >= k)
result += memo[preSum - k];//
}
return result;
}
};