【Lintcode】1844. Subarray Sum Equals K II

题目地址：

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

给定一个数组，再给定一个数字$k$，问该数组中和为$k$的子数组中长度最短是多少。如果不存在这样的子数组则返回$-1$。

思路是用前缀和，用一个哈希表存储前缀和以及其出现的最后位置（因为要求最短的这样的子数组，所以如果相同的前缀和出现了两次，我们需要用后出现的那次）。代码如下：

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 minimum length of continuous subarrays whose sum equals to k
     */
    public int subarraySumEqualsKII(int[] nums, int k) {
        // write your code here
        // key存储前缀和，value存储该前缀和是nums的前多少个数字构成的
        Map<Integer, Integer> preSum = new HashMap<>();
        // 前0个数的前缀和就是0
        preSum.put(0, 0);
        // sum存储当前的前缀和，res记录最终结果
        int sum = 0, res = Integer.MAX_VALUE;
        for (int i = 0; i < nums.length; i++) {
        	// 更新前缀和
            sum += nums[i];
            // 如果该前缀和与k的差出现在前缀和中，说明已经找到了一个和为k的子区间，更新res
            if (preSum.containsKey(sum - k)) {
                res = Math.min(res, i + 1 - preSum.get(sum - k));
            }
            // 否则将该前缀和出现的最新位置存入哈希表
            preSum.put(sum, i + 1);
        }
        
        return res == Integer.MAX_VALUE ? -1 : res;
    }
}

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

算法正确性，只需注意每次res更新时，总是会取到以nums[i]结尾的和为$k$的最短子数组的长度即可。