【Leetcode】560. 和为K的子数组

题目描述

给你一个整数数组 nums 和一个整数 k ，请你统计并返回该数组中和为 k 的连续子数组的个数。

题解

暴力解法，双循环，i指针从左往右走，j指针从i往左走，一个个遍历一个个加起来，直到加到等于k，就计数一次。

执行用时：1445 ms, 在所有 Java 提交中击败了21.59%的用户

内存消耗：41.5 MB, 在所有 Java 提交中击败了27.77%的用户

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

前缀和

执行用时：24 ms, 在所有 Java 提交中击败了56.39%的用户

内存消耗：40.8 MB, 在所有 Java 提交中击败了66.09%的用户

class Solution {
    public int subarraySum(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);

        int preSum = 0;
        int count = 0;
        for (int num: nums) {
            preSum += num;
            if (map.containsKey(preSum - k)) {
                count += map.get(preSum - k);
            }
            map.put(preSum, map.getOrDefault(preSum, 0) + 1);
        }
        return count;
    }
}