LeetCode-560. Subarray Sum Equals K

来源:互联网 发布:java手机编译器aide 编辑:程序博客网 时间:2024/06/01 09:00

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);
原创粉丝点击