[LeetCode] 525、连续数组

题目描述

给定一个二进制数组, 找到含有相同数量的 0 和 1 的最长连续子数组（的长度）。

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

解题思路

这还是一个“计数类”问题，可以用哈希表（map映射）来解决。

这个题的思路不太好想，需要转换一下：把数组中的0看做-1，那么我们可以把问题转化为“求和为0的最长连续子数组（的长度）”。

我们可以利用“前缀和”的思路，定义一个unordered_map用于存放“连续子数组的和”及“其对应的下标”：保存所有的(sum, index)对，对于一个sum，我们只记录它第一次出现的位置，后面遇到相同的 sum，我们都用这个第一次记录的 index 来计算子数组的长度。（“前缀和”的思路）

思路解释：如果当在遍历数组的过程中出现了相同的sum两次，则代表在这两个位置之间（左开右闭）的连续子数组的和为0，也就是说这两个位置之间的 0 和 1 的数目一样多，所以此时我们就可以更新一下最大长度res了。（我的实现）

参考代码

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        unordered_map<int, int> umap = {{0, -1}};  // 用“前缀和”思路时 别忘记初始化
        int sum = 0;
        int res = 0;
        for(int i = 0; i < nums.size(); i++){
            sum += nums[i] == 1? 1: -1;
            if(umap.count(sum)){
                res = max(res, i - umap[sum]);  // 注意此时不更新 umap[sum] = i
            }else{
                umap[sum] = i;
            }         
        }
        
        return res;
    }
    
};