面试题10：和为k的子数组（Java版）

题目：输入一个整数数组和一个整数k，请问数组中有多少个数
字之和等于k的连续子数组？例如，输入数组[1，1，1]，k的值为
2，有2个连续子数组之和等于2。

// 因为求得是一个连续子数组，所以当m大于n时，第前m项的和减去
// 第前n项的和等于k则代表n+1到m之间连续的数组的和等于k以此来
// 求出和为k的连续数字的个数
public static int subArraySum(int[] nums, int k) {
    Map sumToCount = new HashMap<>();
    // 当sum - k等于0时说明当前连续数字本身就满足条件，
    // 所以需要初始哈希表中key为0的value为1
    sumToCount.put(0, 1);
    int sum = 0;
    int count = 0;
    for (int num: nums) {
        sum += num;
        // count加上key为sum - k的键在哈希表中对应的value
        // （即加上了连续数组之和为sum - k的个数）
        count += sumToCount.getOrDefault(sum - k, 0);
        // 将哈希表中对应和的数量加一
        sumToCount.put(sum, sumToCount.getOrDefault(sum, 0) + 1);
    }
    return count;
}