leetcode560.和为K的子数组

// 暴力法
/**
 * 双指针确定一个子数组，穷举所有数组
 */
class Solution {
    public int subarraySum(int[] nums, int k) {
        if (nums == null)  return 0;
        int length = nums.length;
        int count = 0;
        for (int left = 0; left < length; left++) {
            int sum = 0;
            for (int right = left; right < length; right++) {
                sum += nums[right];
                if (sum == k) {
                    count++;
                }
            }
        }
        return count;
    }
}

// 前缀和
/**
 * 根据当前前缀和，在 preSum 中寻找「与之相减 == k」的「目标前缀和」。
 * 目标前缀和这个数值可能不止出现 1 次，假设为 n 次，就等价于，
 * 能找到 n 个子数组的求和等于 k，遍历 nums 数组每一项，n 不断累加给 count，
 * 最后返回 count
 */
class Solution {
    public int subarraySum(int[] nums, int k) {
        if (nums.length == 0) return 0;
        
        // 计算前缀和数组
        int[] preSum = new int[nums.length + 1];
        preSum[0] = 0;
        for (int i = 0; i < nums.length; i++) {
            preSum[i + 1] = preSum[i] + nums[i];
        }

        int count = 0;
        for (int left = 0; left < nums.length; left++) {
            for (int right = left; right < nums.length; right++) {
                // 区间和 [left..right]，注意下标偏移
                if (preSum[right + 1] - preSum[left] == k) {
                    count++;
                }
            }
        }
        return count;
    }
}

// 前缀和哈希表优化
class Solution {
    public int subarraySum(int[] nums, int k) {
         // key：前缀和，value：key 对应的前缀和的个数
         Map<Integer, Integer> preSumFreq = new HashMap<>();
         // 对于下标为 0 的元素，前缀和为 0，个数为 1
         preSumFreq.put(0, 1);

         int preSum = 0;
        int count = 0;
        for (int num : nums) {
            preSum += num;

            // 先获得前缀和为 preSum - k 的个数，加到计数变量里
            if (preSumFreq.containsKey(preSum - k)) {
                count += preSumFreq.get(preSum - k);
            }

            // 然后维护 preSumFreq 的定义
            preSumFreq.put(preSum, preSumFreq.getOrDefault(preSum, 0) + 1);
        }
        return count;
    }
}