【LeetCode热题100(10/100)】和为 K 的子数组

题目地址:链接
题目描述: 给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的子数组的个数 。
子数组是数组中元素的连续非空序列。

示例输出:

输入:nums = [1,1,1], k = 2
输出:2

前缀和+哈希表

思路: 使用哈希表记录前缀和的次数(需要额外加上一个0,即什么都不选)。每次获取另外一个前缀和sum - k是否存在,sum - k == 0

var subarraySum = function(nums, k) {
    let n = nums.length;
    let ans = 0;
    let hash_map = {0: 1};
    let sum = 0;
    for(let i = 0; i < n; i ++) {
        sum += nums[i];
        let key = sum - k;
        if(key in hash_map) ans += hash_map[key] // 记录当前数组和为k的个数
        if(sum in hash_map) hash_map[sum] ++; // 记录当前和为sum的个数
        else hash_map[sum] = 1;
    }
    return ans;
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值