2022-7-22 剑指offer-前缀和+hash

剑指 Offer II 011. 0 和 1 个数相同的子数组

难度中等78收藏分享切换为英文接收动态反馈

给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。

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

思路：0和1可以变成-1和1，转化为和为0的子数组，于是就和之前的和为k的子数组一样，但是要求最长，所以hash记录的是最前面的索引。