算法——0和1个数相同的子数组

请添加图片描述
解题思路:利用前缀和和哈希表

  1. 把输入数组中所有的0都替换成-1,那么题目就变成了求包含相同数目的-1和1的最长子数组的长度
  2. 在一个只包含数字1和-1的数组中,如果子数组-1和1的数目相同的话,那么子数组的所有数字之和就是0
  3. 所以这个题目就变成了求数字之和为0的最长子数组的长度

遍历每一个元素,计算变形过的前缀和,哈希表中存放每个前缀和与该前缀和第一次出现的下标。当该前缀和第二次出现,意味着这两个前缀和的差是0,也就是说在两个前缀和的下表中存在着相同个数的0和1。

class Solution {
    public int findMaxLength(int[] nums) {
        //利用前缀和 和哈希表
        int n = nums.length;
        Map<Integer,Integer> map = new HashMap<>();
        int maxLength =0;
        // 哈希表中存放前缀和,因为第0项的前缀和是0,所以需要先存入
        map.put(0,-1);
        // 前缀和变量
        int persum=0;
        for(int i=0;i<n;i++){
            if(nums[i]==1){
                persum++;
            }else{
                persum--;
            }
            if(map.containsKey(persum)){
                int prevIndex = map.get(persum);
                // 该前缀和在哈希表中已经存在,该前缀和数值第二次出现,意味着,在prevIndex和i之间存在相同个数的0和1;
                maxLength = Math.max(maxLength,i-prevIndex);
            }else{
                // 哈希表中存放前缀和和,该前缀和第一次出现的下标
                map.put(persum,i);
            }
        }
        return maxLength;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值