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的最长连续子数组。

提示：

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

哈希表 中等

class Solution:
    def findMaxLength(self, nums: List[int]) -> int:
        preSum = 0
        counter = {0:-1}
        ans = 0
        for i,n in enumerate(nums):
            if n:
                preSum += 1
            else:
                preSum -= 1
            
            if preSum in counter:
                ans = max(i-counter[preSum],ans)
            elif preSum not in counter:
                counter[preSum] = i
        return ans

关于为什么这样初始化couter

我们假定有一个数组 nums = [0,1]，并且初始化 counter = {}，在遍历到 nums[1] 时，preSum = 0，这时候查表 counter[0] 是不存在的，而长度为2的数组 [0,1] 又符合题意，因此应当初始化 counter = {0:-1}。