【Lintcode】1844. subarray sum equals k II

题目地址:

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

给定一个数组,再给定一个数字kkk,问该数组中和为kkk的子数组中长度最短是多少。如果不存在这样的子数组则返回1-1−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)O(n)O(n)。

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

上一篇:[LintCode 1691.] 买卖股票的最佳时机V


下一篇:Day three