力扣刷题日记

给定一个二进制数组 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 <= 10⁵
nums[i] 不是 0 就是 1

根据题目，数组中只有 0 和 1，要使子数组中 0 和 1的数量相同，那么该子数组中所有元素的和应该为子数组大小的一半。还可以继续简化，即把 0 看成 - 1， 那么子数组中 0 和 1 数量相同时，元素之和为 0。那么这道题可以转化为：找出给定数组中 子数组元素和为 0 最长子数组的长度。可以使用跟上一题相同的方法，即 前缀和 + 哈希表（也可以使用 前缀和 + 双层 for 循环，但是还是同样的问题，时间复杂度高）。代码如下：

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int size = nums.size();
        int ret = 0;
        int total = 0;
        unordered_map<int, int> um;
        um[0] = -1; //key为前缀和， value为该前缀和第一次出现的位置。初始没有算前缀和，所以位置为-1
        for(int i = 0; i < size; ++i){
            total += nums[i] ? 1 : -1;
            if(um.count(total))
                ret = max(ret, i - um[total]);
            else
                um[total] = i;
        }
        return ret;
    }
};

时间复杂度：O(n)；空间复杂度：O(n)。