具有相同数量 0 和 1 的最大子数组

在数组中找出具有相同数量的0和1的最长连续子数组。

例如：

输入：[0,1,0] 输出：2 解释：[0, 1] 是具有相同数量的0和1的最长连续子数组。

输入：[0,1,0,0,1,1,0] 输出：6 解释：[1,0,0,1,1,0] 是具有相同数量的0和1的最长连续子数组。

解法

使用前缀和和哈希表。

具体步骤如下：

1. 将原数组中的0替换为-1，这样当一个子数组的和为0时，0和1个数相等。
2. 遍历数组，计算每个位置的前缀和，并用哈希表存储每个前缀和第一次出现的位置。
3. 遍历哈希表，对于每个前缀和，在哈希表中找到第一次出现的位置和最后一次出现的位置，取二者的差值即可。

代码片段：

class Solution:
    def findMaxLength(self, nums: List[int]) -> int:
        dic = {0:-1}
        max_len = 0
        pre_sum = 0
        for i in range(len(nums)):
            if nums[i] == 0:
                pre_sum -= 1
            else:
                pre_sum += 1
            if pre_sum in dic:
                max_len = max(max_len, i - dic[pre_sum])
            else:
                dic[pre_sum] = i
        return max_len

时间复杂度：O(n)

空间复杂度：O(n)

该解法经过LeetCode测试，时间复杂度和空间复杂度表现都非常优秀。