数组中元素只有1,-1,求和为0的最长子数组

这是算法题未排序数组中累加和为给定值的最长子数组系列问题的一种情况,数组元素只有1,-1,要求和为0的最长子数组。这道题的结果就是比较和为0的子数组长度,哪个最长,结果就是哪一个。例如:

这里最暴力的算法就是:从数组开始到结尾依次做遍历,每一趟遍历中,再计算累加和,结果为0则计算出一个长度,和默认的长度0对比,如果大于默认长度,那么默认长度设置为该长度,一直遍历到数组最后一个元素为止。

算法的代码如下:

这种算法复杂度是O(n*n), 也是最简单明了的做法。

还有一种比较特殊技巧,就是只需要一趟累加,比如:

nums[0],
nums[0]+nums[1],
nums[0]+nums[1]+nums[2],
nums[0]+nums[1]+...+nums[n-1]

虽然我们不能保证这中间的累加和一定会出现0,但是我们可以通过当前的累加和 和 已经计算过的累加和做一个减法,如果这个减法的结果正好是0,那么我们可以认为当前会有一个子数组满足累加和为0。算法的思路如下:

a=a,a+b=a,那么可以认为b=0。

这个算法里面,我们需要一个容器HashMap来保存已经计算过的累加和和下标位置。而且这个map中会有一个默认的元素(0,-1),在这篇文章里的例子中,我们的map会先后存放:(0,-1),(1,0),(2,1),(3,2),(4,3)。

在本例中,通过该算法计算出累加和为2的一个最长子数组就是从开始到倒数第二个元素的位置:

算法的代码如下:

完整代码如下:

package com.xxx.algorithm;

import java.util.HashMap;

public class MaxSubArray {

	public static int maxSub(int[] nums,int k){
		if(nums==null||nums.length==0){
			return 0;
		}
		int length = 0;
		int sum = 0;
		HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
		map.put(0, -1);
		for(int i=0;i<nums.length;i++){
			sum += nums[i];
			if(map.containsKey(sum-k)){
				length = Math.max(i-map.get(sum-k), length);
			}
			if(!map.containsKey(sum)){
				map.put(sum, i);
			}
		}
		return length;
	}
	
	public static int maxSubArray(int[] nums,int k){
		if(nums==null||nums.length==0){
			return 0;
		}
		int sum = 0;
		int length = 0;
		for(int i=0;i<nums.length;i++){
			for(int j=i;j<nums.length;j++){
				sum = sum+nums[j];
				if(sum==k&&(j-i>length)){
					length = j-i+1;
				}
			}
			sum=0;
		}
		return length;
	}
	
	public static void main(String[] args) {
		int[] nums = {1,1,1,1,-1,-1,-1,1,1,-1,1};
		System.out.println("max-length:"+maxSub(nums,0));
	}

}

本例的代码是一个通用的算法代码,他不仅可以计算累加和为0的子数组,还可以计算累加和为任意数k的最长子数组。 

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

luffy5459

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值