题目地址:
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的最短子数组的长度即可。