0
点赞
收藏
分享

微信扫一扫

leetcode 560. Subarray Sum Equals K(和等于K的子数组)

拾杨梅记 2022-02-10 阅读 61

在这里插入图片描述
给出一个整数数组nums和整数k, 返回和为k的子数组的个数。
子数组是连续的。

思路:
采用积分数组的思想,就是说sum[i] = nums[0] + nums[1] + … + nums[i]
那么从i 到 k这部分子数组的和就等于sum[k] - sum[i]

子数组的和等于k,就相当于sum[k] - sum[i] = k,
那么sum[k] - k = sum[i]
是不是转换成了Two sum的问题呢

区别是这里nums的元素存在负数,就是说可能存在重复的和,所以要用hashMap, key是sum,value是sum出现的次数。
遍历数组,每个元素处求sum,装入hashMap,map中已存在sum时,value加1。
当hashMap中存在sum-k时,结果次数相加。

注意key=0时,value=1,就是不加任何元素的情况。

    public int subarraySum(int[] nums, int k) {
        HashMap<Integer, Integer> map = new HashMap<>();
        int n = nums.length;
        int sum = 0;
        int result = 0;
        
        map.put(0, 1);
        for(int i = 0; i < n; i ++) {
            sum += nums[i];
            if(map.containsKey(sum - k)) {
                result += map.get(sum - k);
            }
            map.put(sum, map.getOrDefault(sum, 0) + 1);
        }
        
        return result;
    }
举报

相关推荐

0 条评论