LeetCode 560. Subarray Sum Equals K （子数组之和等于K）

Posted 2020-10-09 几米空间

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:

1. The length of the array is in range [1, 20,000].
2. The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].

 

 

---

题目标签：Array, Map

　　题目给了我们一个nums array 和一个 k， 要我们找出有多少个子数组 之和是等于 k的。

　　一开始想到的是暴力解法，虽然能够通过，但是n*n 始终是太慢。

　　这题可以利用HashMap，把出现过的sum 当作key 存入， 把这个sum 出现过的次数 当作value 存入。

　　遍历nums array，一直更新sum，然后去map 里找有没有 sum - k，有的话说明 sum - k 是一个旧的sum，之前出现过。换句话说，新的sum - 旧的sum = k，说明 新的sum 减去 旧的sum，剩下的那一段的 和 等于k， 找到了一个子数组之和 = k的。然后在把新的sum 存入 map里。

　　这题的关键就是，当我们知道了一个 sum（0，i） 和 sum（0，j）的话，我们也就知道了 sum（i+1， j）。

　　所以当我们知道了目前所有旧的sum， 当我们有了一个新的sum，我们可以利用map 去找到 新的sum - k 是不是在map里存在。一旦存在，说明了我们找到了一个 子数组它的和为k。

　　要注意的是，这里count +的是map 里的value，而不是count++， 因为当你在map 里找到了一个 sum - k的旧sum的时候，这个旧的sum 可能出现过2次，换句话说，可能有两个长度不一样的子数组，但是它们的和都等于 sum - k（因为有负数）。所以这里要加上旧sum 的出现次数2，因为你找到了两个不同的子数组它们的和都为k。

 

 

Java Solution:

Runtime beats 67.25% 

完成日期：10/03/2017

关键词：Array, HashMap

关键点：把所有出现过的sum存入map：sum为key，出现的次数为value；利用map来找k (新sum - k =? 任何旧sum)  

 1 class Solution 
 2 {
 3     public int subarraySum(int[] nums, int k) 
 4     {
 5         int count = 0;
 6         int sum = 0;
 7         
 8         Map<Integer, Integer> map = new HashMap<>();
 9         map.put(0, 1); // initial value sum = 0, occurrence = 1 for case sum = k, k - k = 0 counts
10         
11         for(int i=0; i<nums.length; i++) // iterate nums array
12         {
13             sum += nums[i]; // update sum
14             
15             if(map.containsKey(sum - k)) // if map has sum - k, meaning this new sum - old sum = k
16                 count += map.get(sum - k); // previous sum might appear more than once
17                                            // this is why we need to add its value
18             map.put(sum, map.getOrDefault(sum, 0) + 1); // save new sum into map
19         }
20         
21         return count;
22     }
23 }

参考资料：

https://discuss.leetcode.com/topic/87850/java-solution-presum-hashmap

 

LeetCode 题目列表 - LeetCode Questions List