leetcode 0 和 1 个数相同的子数组

题目链接
思路:前缀和+hash表
分析:数组中只有0和1两个数字,那么将数组前缀和的意思就表示到当前位置为止,一共有多少个1,因为只有0和1两个数字,那么当前位置-当前位置1的个数+1就是到当前位置一共有多少个0.再将遍历每个位置,1的个数比0的个数多几个。
简单说一下上面的话:
1.统计当前位置为止,1的个数
2.统计当前位置为止,0的个数
3.统计当前位置为止,1的个数比0的个数多几个(存在tar数组中)

然后遍历tar数组
1.看当前位置是否等于0,如果等于0,那么说明当前位置为止,1和0的个数相等。更新res,并且把0此时的当前为止放在hash表中
2.查看hash表中是否存在当前1和0 个数的差值,如果存在,说明当前为止到之前那个位置,中间1和0的个数是一样的。更新res,此时不用把当前这个差值在hash表中的值更新。
3.否则,直接放到hash表中
代码:

class Solution {
    public int findMaxLength(int[] nums) {
        int len = nums.length;
        //1的个数
        int[] onePresum = new int[len];
        onePresum[0] = nums[0];
        for(int i = 1;i <len;i++){
            onePresum[i] = nums[i] + onePresum[i-1];
        }
		//0的个数
        int[] zeroPresum = new int[len];
        for(int i = 0;i < len;i++){
            zeroPresum[i] = i+1 - onePresum[i];
        }
		
		//记录1的个数比0的个数多几个
        int[] tag = new int[len];
        for(int i = 0 ; i < len;i++){
               tag[i] = onePresum[i] - zeroPresum[i];
        }
		
		// key:差值   value:出现的下标
        HashMap<Integer, Integer> map = new HashMap<>();
        int res = 0;
        for(int i = 0; i< len;i++){

            if(tag[i] == 0){
                res = Math.max(res,i+1);
                if(!map.containsKey(0)){
                    map.put(0,i);
                }
            }else if(map.containsKey(tag[i])){
                res = Math.max(res, i - map.get(tag[i]));
            }else{
                map.put(tag[i],i);
            }

        }
        return res;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值