和为K的子数组

问题:
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。

示例 :
输入:nums = [1,1,1],k = 2
输出: 2。[1,1] 与 [1,1] 为两种不同的情况。

说明 :
数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。


分析:
对于该问题,最简单直接的方法就是暴力法。将每个子数组的和与目标值 k 比较,得到结果。(暴力法就不过多介绍了)

代码如下:

public int subarraySum(int[] nums, int k) {
	int length = nums.length, count = 0;
    for(int i = 0; i < length; i++) {
        int sum = 0;
        for(int j = i; j < length; j++) {
            sum += nums[j];
            if(sum == k )
                count++;
        }
    }
    return count;
}

下面介绍一个新的题解:

我们假设有 [a, b] 是一个满足条件的区间,即有:数组的前 b 项和减去前 a 项和为目标值 k,sum(b) - sum(a) = k,即:前 b 项和减去目标值 k,得到前 a 项和,sum(b) - k = sum(a)。

这里就是重点了,我们用前 b 项和减去目标值 k,得到前 a 项和,若前 a 项和不存在,即找不到满足条件的区间 [a, b];若前 a 项和存在,则区间满足,将前 a 项和出现的次数,累加在 count 中。

我们需要一个 Map<Integer, Integer> sumAndCount,将每个区间和作为 key,区间和出现的次数作为 value,保存到 sumAndCount 中,若要保存的区间和已经存在,次数设置为 当前和出现次数 + 1,若不存在,次数设置为 1。

代码如下:

public int subarraySum(int[] nums, int k) {
    int count = 0,sum = 0;
    Map<Integer, Integer> sumAndCount = new HashMap<>();
    // (0, 1) 是因为可能存在前 n 项和为 k 的情况,k - k = 0
    sumAndCount.put(0,1);
    for(int t : nums) {
    	// sum 保存第一项到当前项的和
        sum += t;
        // 判断是否存在某项和为 sum - k
        // 若存在,将 sum - k 出现的次数累加到 count 中
        if(sumAndCount.containsKey(sum-k))
            count += sumAndCount.get(sum-k);
        // 将当前和存到 sumAndCount 中
        sumAndCount.put(sum, 1+sumAndCount.getOrDefault(sum,0));
    }
    return count;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值