Leetcode: Maximum Size Subarray Sum Equals k


Given an array nums and a target value k, find the maximum length of a subarray that sums to k. If there isn't one, return 0 instead.

Example 1:
Given nums = [1, -1, 5, -2, 3], k = 3,
return 4. (because the subarray [1, -1, 5, -2] sums to 3 and is the longest)

Example 2:
Given nums = [-2, -1, 2, 1], k = 1,
return 2. (because the subarray [-1, 2] sums to 1 and is the longest)

Follow Up:
Can you do it in O(n) time?

Like the other subarray sum problems Lintcode: Subarray Sum closest

Use a HashMap to keep track of the sum from index 0 to index i, use it as the key, and use the current index as the value

build the hashmap: scan from left to write, if the current sum does not exist in the hashmap, put it in. If the current sum does exist in Hashmap, do not replace or add to the older value, simply do not update. Because this value might be the left index of our subarray in later comparison. We are looking for the longest subarray so we want the left index to be the smaller the better. 

Every time we read a number in the array, we check to see if map.containsKey(num-k), if yes, try to update the maxLen.

 1 public class Solution {
 2     public int maxSubArrayLen(int[] nums, int k) {
 3         if (nums==null || nums.length==0) return 0;
 4         HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
 5         map.put(0, -1);
 6         int sum = 0;
 7         int maxLen = Integer.MIN_VALUE;
 8         for (int i=0; i<nums.length; i++) {
 9             sum += nums[i];
10             if (!map.containsKey(sum)) {
11                 map.put(sum, i);
12             }
13             if (map.containsKey(sum-k)) {
14                 int index = map.get(sum-k);
15                 maxLen = Math.max(maxLen, i-index);
16             }
17         }
18         return maxLen==Integer.MIN_VALUE? 0 : maxLen;
19     }
20 }

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM