leetcode560.和为K的子数组

题目来源：题目

题目

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

 

示例 1 :

 

输入:nums = [1,1,1], k = 2

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

说明 :

 

数组的长度为 [1, 20,000]。

数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

 

思路

思路一（枚举）

计算每个以下标j结尾的子数组和，判断该子数组的和是否为k。下标为j的子数组有j+1个，但是计算子数组的和时从后往前计算就比较简单，比如知道了子数组[i,j]的和，那么[i-1,j]的和就是[i,j]的和加上nums[i-1]。

思路二（前缀和+哈希表）

       我们把下标为从0-j的数组和叫做j个前缀和。记作pre[j]。

       假如下标从i-j的子数组和为k。就有pre[j] – pre[i – 1] == k.

       即pre[i-1] == pre[j] – k 这个公式中有0 <= i <= j

       于是我们可以计算每个pre[j] 的值，然后判断之前计算的前缀和中pre[j] – k的值有多少个，就是和为k的子数组个数。

       可以用哈希表的键来存储pre[j],用哈希表的键值对应前缀和为pre[j]的个数。这样在查询的时候时间复杂度就为O（1）。

代码

思路一代码

public static int subarraySum(int[] nums, int k) {
    int result = 0;
    for(int end = 0;end < nums.length;end++){
        int sum = 0;
        for(int start = end;start >= 0;start--){
            sum += nums[start];
            if(sum == k)result++;
        }
    }
    return result;
}

思路二代码

public static int subarraySum(int[] nums, int k) {

    int pre = 0,result = 0;

    Map<Integer,Integer> mp = new HashMap<>();

    mp.put(0,1);

    for(int i = 0;i < nums.length;i++){

        pre += nums[i];

        if(mp.containsKey(pre - k))result += mp.get(pre - k);

        mp.put(pre,mp.getOrDefault(pre,0)+1);

    }

    return result;

}

总结

思路一的时间复杂度为O(n^2),空间复杂度为O（1）

思路二的时间复杂度为O(n),空间复杂度为O(n)