这一题的follow up要求是复杂度O(N),很抱歉,我不知道O(n)以外的做法。这题的解法的关键点主要是以下两点:
1. sum[m .. n] = sum[1..n] - sum[1..m - 1]
2. k = sum[m .. n] = sum[1..n] - sum[1..m - 1] => sum[1..n] - k = sum[1..m - 1]
所以其实整个算法流程就是这样的。
不停往下走,记录当前的和,然后记录下来当前和所对应的位置,如果当前和之前出现过了就不记录了,因为你需要的是maximum size,所以m是越小越好。然后遍历的同时寻找是否存在sum[1..i] - k,如果存在,那个对应的位置就会构成一个subarray sum equals K。 然后找到其中最长的就可以了。根据这个算法,给出代码如下
public int maxSubArrayLen(int[] nums, int k) {
HashMap<Integer, Integer> cacheMap = new HashMap<>();
int result = 0;
int curSum = 0;
cacheMap.put(0, -1);
for (int i = 0; i < nums.length; i++) {
curSum += nums[i];
if (!cacheMap.containsKey(curSum)) {
cacheMap.put(curSum, i);
}
int candidate = curSum - k;
if (cacheMap.containsKey(candidate)) {
int loc = cacheMap.get(candidate);
if (loc == i) continue;
result = Math.max(result, i - loc);
}
}
return result;
}