2021-06-03 力扣每日一题

525 连续数组

问题描述：

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

示例1：

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

示例2：

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

思路：

【前缀和+哈希表】

​ 最近都是前缀和+哈希表的问题。其实拿到这个问题最初想的是滑窗，毕竟是连续的嘛，所以滑窗是一定能做出来的，但一般滑窗是n的三次方，优化过后也是n²，显然是不可以的，所以这里我们可以使用前缀和的思想。

​ 我们先计算出数组的前缀和，因为这个集合的自由度是2，所以如果进行遍历的话，时间复杂度依然是n²级别的，所以这里我们需要引入哈希表进行操作。

​ 遍历到某个位置，如果这个位置的前缀和与之前某个位置的前缀和是相同的，那么他们之间的和就是0，这样我们就找到一个这样的区间。那么我们只需要保证这个区间最大即可。这样我们就可以在n的复杂度下，将问题解决

​ 但是，实现这个操作需要一个前提，那就是我们需要将所有的0看成-1，这样如果存在相同数量的0和1的话，这个区间的和就是0。有了这样的思路，我们就有了代码的思路，首先应该先将初始值输入，接着遍历即可。

Java代码

/**
* @Description: 力扣525题题解
* @return:  返回结果
* @Author: Mr.Gao
* @Date: 2021/6/3
*/
public int findMaxLength2(int[] nums) {
    Map<Integer,Integer> map = new HashMap<>();
    map.put(0, -1);
    int n = nums.length;
    int s = 0;
    int re = 0;
    for (int i = 0; i < n; i++) {
        if(nums[i]==1){
            s++;
        }else{
            s--;
        }
        if(map.containsKey(s)){

            re = Math.max(i-map.get(s),re);
        }else{
            map.put(s,i);
        }
    }
    return re;
}