LeetCode每日十题---哈希表（三）

最新推荐文章于 2021-07-25 15:25:52 发布

原创最新推荐文章于 2021-07-25 15:25:52 发布 · 641 阅读

0 ·

CC 4.0 BY-SA版权

leetcode刷题专栏收录该内容

37 篇文章

订阅专栏

本文介绍了一种高效的方法来寻找一个数组中最长的含有等数量0和1的子数组。通过将0替换为-1，问题转化为寻找元素和为0的最长子数组。使用哈希表记录每个和首次出现的位置，可以在线性时间内解决此问题。

题目描述1

在这里插入图片描述

笔者解答1.1

class Solution {
    public int findMaxLength(int[] nums) {
       int res=0,sum=0;
       for(int i=0;i<nums.length;i++){
           if(nums[i]==0){
               nums[i]=-1;
           }
       }
       Map<Integer,Integer> map=new HashMap<>();
       for(int i=0;i<nums.length;i++){
           sum+=nums[i];
           if(sum==0){
               res=i+1;
           }
           if(map.containsKey(sum)){
               res=Math.max(i-map.get(sum),res);
           }else{
               map.put(sum,i);
           }
       }
       return res;
    }
}