比较简单易懂的优美子数组 C++ HashMap(2020/4/21)

给你一个整数数组 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;
    }
};
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值