560. 和为K的子数组

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int sum = 0, res = 0;
      	// 注意这里前缀和多了一个0，防止漏掉数组的前缀和刚好等于k的情况
        unordered_map<int, int> um{{0, 1}};
        for (int& num : nums) {
            sum += num;
            res += um[sum - k];
            ++um[sum];
        }
        return res;
    }
};

天真的我以为学会了前缀和就可以了，但是简单的前缀和还是会有O(N^2)的时间复杂度。直接抄了代码，是前缀和+hash表的方法。
定义一个unordered_map，因为只遍历一次，而且有hash表，所以不用前缀表数组了，只需一个int即可，每次把sum加上nums[i]，res就加上um[sum-k]，如果sum-k那一位不为0，证明从sum-k到sum期间，是有组合的和为k的。最后将um[sum]++.
为什么um[0]==1，因为防止sum-k=0，也就是sum刚好=7时漏掉