?博主介绍

🌊 作者主页：苏州程序大白

🌊 作者简介：🏆CSDN人工智能域优质创作者🥇，苏州市凯捷智能科技有限公司创始之一，目前合作公司富士康、歌尔等几家新能源公司

💬如果文章对你有帮助，欢迎关注、点赞、收藏

💅 有任何问题欢迎私信，看到会及时回复
💅关注苏州程序大白，分享粉丝福利

0 和 1 个数相同的子数组

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

示例 1：
输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量 0 和 1 的最长连续子数组。

示例 2：
输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量 0 和 1 的最长连续子数组。

提示：

1 <= nums.length <= 105
nums[i] 不是 0 就是 1

解题思路

错误的思路：滑动窗口，双指针

一开始打算运用双指针的想法，但是就实际而言是难以行通的。

对于[0,0,0,1,1,1,0]最后可能为误判为0。但是实际是6

总结一下，当不能明确得知可以淘汰的某些值时候就最好不要使用双指针。

前面那些0并不能直接判断是否需要淘汰,需要根据后面很多数来确定

public int findMaxLength(int[] nums) {
    int res=0;
    int n=nums.length;
    int slow=0, fast=0;
    int zeroNum=0, oneNum=0;
    
    while (fast<n){
        for (int i=0; i<2 && fast<n; i++){
            if (nums[fast++]==0) zeroNum++;
            else oneNum++;
        }
        if (oneNum==zeroNum) res = Math.max(res, fast-slow);
        else {
            if (nums[slow++]==0) zeroNum--;
            else oneNum--;
        }
    }

    return res;
}

前缀和+哈希表

把0当做-1处理，这里就转变为了求和为0的子数组的问题
key=前缀和，value=第一次出现过的位置
遍历数组，记录某个前缀和第一次出现的位置，设置初始值为map[0] = -1
当前缀和sum在此前出现过，则说明这两个前缀和区间差构成的所有元素和为0，满足条件更新res的值，否则记录到map中

在这里插入图片描述

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