LeetCode-560. Subarray Sum Equals K

Given an array of integers and an integer k, you need to find the total number of continuous subarrays whose sum equals to k.

Example 1:
Input:nums = [1,1,1], k = 2
Output: 2
Note:
The length of the array is in range [1, 20,000].
The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].

方法一 ：暴力搜索

对于每个子数组，分别计算出它们的数组和，再与所给的k值比较是否相等。
时间复杂度：O(n^3)
空间复杂度：O(1)
由题目中note出的数组长度和数组中元素的跨度可知暴力搜索必然会超时，这个思路直接就pass掉了。

方法二

对于数组中的每个元素，不断叠加计算从此元素往后每个子数组的和，遇到子数组的和与k相等，则将计数器加一。每当这一过程结束后，需要重新将用来保存和的变量初始化为0。
时间复杂度：O(n^2)
空间复杂度：O(1)

class Solution {
    public int subarraySum(int[] nums, int k) {
        int count=0;
        for(int start=0;start<nums.length;++start)
        {
            int sum=0;
            for(int end=start;end<nums.length;++end)
            {
                sum+=nums[end];
                if(sum==k)
                    count++;
            }
        }
        return count;
    }
}

方法三

此方法需要用到Map集合，思路如下
1. 每遍历到一个数时就把当前叠加的和作为key加入Map集合中，value值为1,；如果已经存在，则value+1。
2. 如果当前和减去k所得值存在于Map集合中，说明至少存在一个子数组和为k，而具体有多少个子数组需要看Map集合中的value值。
时间复杂度：O(n)
空间复杂度：O(n)

class Solution {
    public int subarraySum(int[] nums, int k) {
        int count=0,sum=0;
        Map<Integer,Integer> sumMap=new HashMap<>();
        sumMap.put(0,1);
        for(int i=0;i<nums.length;++i)
        {
            sum+=nums[i];
            if(sumMap.containsKey(sum-k))
                count+=sumMap.get(sum-k);
            if(sumMap.containsKey(sum))
                sumMap.put(sum,sumMap.get(sum)+1);
            else
                sumMap.put(sum,1);
        }
        return count;
    }
}

代码改进点

if(sumMap.containsKey(sum))
                sumMap.put(sum,sumMap.get(sum)+1);
            else
                sumMap.put(sum,1);

这段代码可以用JDK8中新增的方法getOrDefault代替，具体如下

sumMap.put(sum,sumMap.getOrDefault(sum,0)+1);