LeetCode 560. Subarray Sum Equals K (子数组之和等于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:

  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

 

posted @ 2017-10-05 00:16  Jimmy_Cheng  阅读(588)  评论(0编辑  收藏  举报