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);