描述
给定一个无序数组arr, 其中元素可正、可负、可0。给定一个整数k,求arr所有连续子数组中累加和为k的最长子数组长度。
保证至少存在一个合法的子数组。
示例1
输入:
[1,-2,1,1,1],0
复制返回值:
3
假设s(i)是子数组arr[0…i]的累加和,那么s(j)就代表arr[0…j]的累加和,那么可求得arr[j+1…i]=s(i)-s(j)。
流程:
1.初始化变量sum=0,表示从arr[0]一直加到arr[i]的累加和。初始化变量len,表示累加到和为k的最长子数组长度。设置一个map,用来记录出现过的sum值和的它出现的最早位置,即第一次出现的位置。key代表出现过的sum,value代表第一次出现的位置。
令sum+=arr[i],即s(i),然后在map中查看是否出现过sum-k。
2.1 如果sum-k存在,则取出它首次出现的位置,假设为j,表示arr[0…j]=sum-k=s(j)。而根据之前的假设arr[j+1…i]=s(i)=s(j),此时sum=s(i),所以arr[j+1…i]=sum-(sum-k)=k。因为map的value记录的是最早出现的位置,所以此时arr[j+1…i]是以arr[i]结尾的所有子数组中,累加和为k的子数组中最长的,然后更新len。
2.2 如果sum-k,不存在,说明以arr[i]结尾的子数组中没有累加和为k的子数组
继续重复2,直到遍历完。
public static int maxLength(int[] arr, int k) {
if (arr == null || arr.length == 0) {
return 0;
}
Map<Integer, Integer> map = new HashMap<>();
map.put(0, -1);
int len = 0;
int sum = 0;
for (int i = 0; i < arr.length; i++) {
sum += arr[i];
if (map.containsKey(sum - k)) {
len = Math.max(len, i - map.get(sum - k));
}
if (!map.containsKey(sum)) {
map.put(sum, i);
}
}
return len;
}