累加和为给定值的最长子数组长度

题目描述

给定一个无序数组arr，其中元素可正、可负、可0，给定一个整数k。求arr所有的子数组中累加和为k的最长子数组长度。

题解

• s(i)=arr[0…i]所有元素的累加和；
  又arr[0…j-1]+arr[j…i]=arr[0…i]；
  所以arr[j…i]=s(i)-s(j-1)，这个结论是求解这道题的核心。

• 设置变量sum=0，表示从0位置开始一直加到i位置所有元素的和。设置变量len=0，表示累加和为k的最长子数组长度。设置哈希表map，其中key表示从arr最左边开始累加的过程中出现过的sum值，对应的value值则表示sum值最早出现的位置

• 从左到右开始遍历，遍历的当前元素为arr[i]

• 1、令sum=sum+arr[i]，即之前所有元素的累加和S(i)，在map中查看是否存在sum-k。

  • 如果sum-k存在，从map中取出sum-k对应的value值，记为j，j代表从左到右不断累加的过程中第一次加出sum-k这个累加和的位置。
    • 根据之前得出的结论，arr[j+1…i]的累加和为s(i)-s(j)，此时s(i)=sum,s(j)=sum-k,所以arr[j+1…i]的累加和为k。同时因为map中只记录每一个累加和最早出现的位置，所以此时的arr[j+1…i]是在必须以arr[i]结尾的所有子数组中，最长的累加和为k的子数组，如果该子数组的长度大于len，就更新len。
  • 如果sum-k不存在，说明在必须以arr[i]结尾的情况下没有累加和为k的子数组。

• 2、检查当前的sum（即s(i)）是否在map中。如果不存在，说明此时的sum值是第一次出现的，就把(sum,i)加入map中。如果sum存在，说明之前已经出现过sum，map只记录一个累加和最早出现的位置，所以此时什么记录也不添加

• 3、继续遍历下一个元素，直到所有的元素遍历完。

• 还有一个重要的问题要注意。arr[j+1…i]=s(i)-s(j)，如果从0位置开始累加，会导致j+1>=1。也就是数，所有从0位置开始的子数组都没有考虑过。所以，应该从-1位置开始累加，也就是在遍历之前先把（0，-1）这个记录放进map，这个记录的意思是如果任何一个数也不加时，累加和为0。

  • 比如arr=[1,2,3,3],k=6，如果从0位置开始累加，也就是遍历之前不加入（0，-1）这条记录，那么当遍历到第一个3时，sum=6，map中记录为（1,0）（3,1），此时sum-k=6-6=0，当在map中查询累加和0最早出现的位置，，发现没有出现过，那么子数组[1,2,3]就被我们忽略。所以遍历之前需要先加入（0，-1）这条记录。

public 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 sum=0,len=0;
		for(int i=0;i<arr.length;i++) {
			sum+=arr[i];
			if(!map.containsKey(sum))
				map.put(sum, i);
			if(map.containsKey(sum-k))
				len=Math.max(len, i-map.get(sum-k));
		}
		return len;
	}

补充题目

1、给定一个无序数组arr，其中元素可正、可负、可0。求arr所有的子数组中正数与负数个数相等的最长子数组的长度。

• 将数组中的所有正数转化为1，负数转化为-1，求和为0的最长子数组长度即可。

2、给定一个无序数组arr，其中元素只是1或0。求arr所有的子数组中0和1个数相等的最长子数组长度

• 将数组中的0全部变成-1，然后求累加和为0的最长子数组长度即可。