Q:给定一个数组 nums 和一个目标值 k,找到和等于 k 的最长子数组长度。如果不存在任意一个符合要求的子数组,则返回 0。
注意:
nums 数组的总和是一定在 32 位有符号整数范围之内的。
示例 1:
输入: nums = [1, -1, 5, -2, 3], k = 3
输出: 4
解释: 子数组 [1, -1, 5, -2] 和等于 3,且长度最长。
示例 2:
输入: nums = [-2, -1, 2, 1], k = 1
输出: 2
解释: 子数组 [-1, 2] 和等于 1,且长度最长。
进阶:
你能使时间复杂度在 O(n) 内完成此题吗?
A:
- 我觉得我有毛病,我竟然第一时间用递归……不过就算用了memo,也超时了
private int len = 0;
private Set<String> set;
public int maxSubArrayLen(int[] nums, int k) {
if (nums.length == 0)
return 0;
set = new HashSet<>();
sum(nums, k, 0, 0, nums[0]);
return len;
}
private void sum(int[] nums, int k, int left, int right, int count) {
String s = left + "+" + right;
if (set.contains(s) || right < left)
return;
set.add(s);
if (count == k) {
len = Math.max(len, right - left + 1);
}
sum(nums, k, left + 1, right, count - nums[left]);
if (right + 1 < nums.length)
sum(nums, k, left, right + 1, count + nums[right + 1]);
}
- 前缀和+map
计算出每个索引的前缀和,利用HashMap储存每个前缀和和对应的索引,如果出现前缀和相同的情况,则储存较小的索引(因为要求最长子数组);利用一个指针i作为子数组的结尾从后向前遍历(从前往后要判断索引位置在当前位置的前还是后,从后往前无需判断,因为如果在后为负),寻找map中是否储存有key为sum[i] - k的索引,如果有则更新max。当指针的值小于等于max的值后,则无需再继续遍历。
public int maxSubArrayLen(int[] nums, int k) {
if (nums.length == 0)
return 0;
int maxLen = 0;
int[] sums = new int[nums.length + 1];
sums[0] = 0;
Map<Integer, Integer> map = new HashMap<>();
map.put(0, 0);
for (int i = 1; i <= nums.length; i++) {
sums[i] = sums[i - 1] + nums[i - 1];
if (!map.containsKey(sums[i]))
map.put(sums[i], i);
}
for (int i = sums.length - 1; i > maxLen; i--) {
if (map.containsKey(sums[i] - k)) {
maxLen = Math.max(maxLen, i - map.get(sums[i] - k));
}
}
return maxLen;
}