【Lintcode】838. Subarray Sum Equals K

题目地址：

https://www.lintcode.com/problem/subarray-sum-equals-k/description

给定一个数组，返回其所有和等于给定数字$k$的子数组的个数。

思路是先求前缀和sum，然后一边遍历sum，一边将前缀和为sum的子数组个数更新到一个哈希表里；当遍历到$sum[i]$，就在哈希表里check一下$sum[i]-k$这个前缀和出现了多少次，每出现一次说明存在一个子数组和为$k$，于是计数res加一。代码如下：

import java.util.HashMap;
import java.util.Map;

public class Solution {
    /**
     * @param nums: a list of integer
     * @param k: an integer
     * @return: return an integer, denote the number of continuous subarrays whose sum equals to k
     */
    public int subarraySumEqualsK(int[] nums, int k) {
        // write your code here
        Map<Integer, Integer> map = new HashMap<>();
        // 算前缀和。sum[i]表示nums前i个数的前缀和。注意，前0个数的前缀和规定是0
        int[] sum = new int[nums.length + 1];
        for (int i = 0; i < nums.length; i++) {
            sum[i + 1] = sum[i] + nums[i];
        }
        
        int res = 0;
        for (int i = 0; i < sum.length; i++) {
        	// 如果前缀和sum[i] - k出现过x次，
        	// 说明以nums[i - 1]结尾的和为k的子数组个数就是x，于是加进res
            if (map.containsKey(sum[i] - k)) {
                res += map.get(sum[i] - k);
            }
            // 接着把前缀和出现次数更新
            map.put(sum[i], map.getOrDefault(sum[i], 0) + 1);
        }
        
        return res;
    }
}

时空复杂度$O(n)$。