leetcode560. 和为K的子数组

传送门

题目：给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。输入:nums = [1,1,1], k = 2 输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

这个问题可以通过Prefix sum来求解。
假设我们令P[i] = A[0] + A[1] + … + A[i] 和 P[j] = A[0] + A[1] + … + A[j]，
那么P[j] - P[i] = A[i] + A[i+1] + … + A[j]。
如果P[j] - P[i] == S的话，那么（i,j] 就是我们需要的区间。
所以我们对于每个j，我们只要计算有多少个i使得 P[j] - P[i] == S，这样我们就得到了以P[j]作为右区间并且和为S的区间数。对于A中的每个元素都做同样的处理，最有将所有的结果相加即可。

具体实现上，我们通过hash_map记录P[j]。

初始化的时候要注意一个细节，对于map[0]=1。为什么？

因为当P[j]==S时，P[i]=0并且此时我们的result=1。

C++:

	int subarraySum(vector<int>& nums, int k) {
        if (nums.size() < 1 || k < -1e7 || k > 1e7)
            return 0;
        // hash表来存储从0到i的序列和
        unordered_map<int, int> map;
        map[0] = 1; // sum本身等于k
        // 结果，0到索引i的序列数值和
        int ans = 0, sum = 0;
        for (int i = 0; i < nums.size(); ++i) {
            sum += nums[i];
            // 如果当前和减去k的值已经在之前存在过，那么加上存在的个数
             // 为什么是sum-k? 因为两个序列和相减才得到一个区间和，k就是区间和
            if (map.count(sum - k) != 0) {
                ans += map[sum - k];
            }
            // sum的个数+1（unordered_map默认初值时0）
            ++map[sum];
        }
        return ans;
	}

Java:

 	public int subarraySum(int[] nums, int k) {
        int ans = 0;
        int sum = 0;
        Map<Integer, Integer> hash = new HashMap<>();
        hash.put(0, 1);
        for (int i = 0; i < nums.length; ++i) {
            sum += nums[i];
            ans += hash.getOrDefault(sum - k, 0);
            hash.put(sum, hash.getOrDefault(sum, 0) + 1);
        }
        return ans;
    }