求一个只包含0,1,-1的数组,和为0的最长子串的长度

转自百度贴吧 @我想去光的深处

求一个只包含0,1,-1的数组,和为0的最长子串的长度;要求时间复杂度为O(n),
例如:{0,-1,0,1,-1,0,1,-1,-1,1} 最长子串长度为8

思路:记录前n项的和
若为零的话len加一,循环继续
为其他值的话,记录第一次出现的位置,若已经记录过,现在的位置减去第一次出现的位置极为len的大小,然后每一次得到的len与上一次比较,
则循环最后一次得到的len即为要求的长度值

HashMap<Integer ,Integer> map = new HashMap();
map.put(0, -1);
int len=0;
int sum=0;
for (int i = 0; i < arr1.length; i++) {
sum+=arr1[i];
if(map.containsKey(sum)){
len=Math.max(i-map.get(sum), len);
System.out.println("sum:"+sum+" len:"+len);
}
if(!map.containsKey(sum)){
map.put(sum, i);
}
}

return len;
点赞