Leetcode560 和为 K 的子数组

题目描述

我想到的是暴力破解，见代码，外层遍历数组，内层以该元素为结尾，累加之前的，和为k，结果加1。这样不会重也不会漏

代码：

public int subarraySum(int[] nums, int k) {
        int res = 0;
        for(int i = 0; i < nums.length; i++){
            int j = i;
            int sum = 0;
            for(; j >= 0; j--){
                sum += nums[j];
                if(sum == k){
                    res++;
                }
            }
        }
        return res;
    }

时间复杂度O(n2)，空间复杂度O(1)

解法二：

本题采用哈希表存储从数组第一个元素不断往后的子序列和，然后判断到当前元素的序列总和减去K的值在哈希表中有多少个，即为包含当前元素的子序列可以得到目标结果，利用前后子序列的差可以得到目标子序列和为K

分析：

【j…i】子数组和为k等价于pre[i]−pre[j−1]==k。即pre[j−1]==pre[i]−k。即考虑以i结尾的和为k的连续子数组时，只需要统计有多少个前缀和为pre[i]-k的pre[j]即可。因为prej是从第一个开始计算的。如此方便计算。因此建立哈希表，以和为键，出现次数为值。
注意：哈希表中是储存（sum，v）和（sum-k，v）的。把这两个变量的值都存在了哈希表中。找的是后者。

代码：

public int subarraySum(int[] nums, int k) {     
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int sum = 0, result = 0;

        for(int i = 0; i < nums.length; ++i) {
            sum += nums[i];
            if(map.containsKey(sum-k))
                result += map.get(sum-k);
            map.put(sum, map.getOrDefault(sum, 0)+1);
        }
        
        return result;
    }

Map.getOrDefault(Object key, V defaultValue)方法的作用是：
  当Map集合中有这个key时，就使用这个key值；
  如果没有就使用默认值defaultValue