525. 连续数组

525. 连续数组

原始题目链接：https://leetcode-cn.com/problems/contiguous-array/

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

示例 1:

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

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

解题思路：

需要一个简单的数学推导，如果在数组中的两个位置x和y，以它们的位置为终点，各自1和0的数量的差值相同，记为k，那么假设x的位置是位于y之前的，即左边，那么假设在x这个位置有m个0，那么1有m + k个，假设y有n个0，那么y有n + k个1，那么在区间[x, y]有n - m个0,1的数量=n + k - m -k = n - m，所以在这个区间里，0和1的梳理是相同的，都是n-m，所以我们就希望这个区间越大越好，这样可以获得最长的连续子数组。

代码实现：

class Solution:
    def findMaxLength(self, nums: List[int]) -> int:
        # 定义一个前缀和字典，key为当前数组中某个位置以i为终点的1和0的个数差值，value为这个位置在数组中的索引
        # 初始化key为0，value为-1
        nums_hash = {0 : -1}

        # 当前1和0个数的差值，初始化为0
        counter = 0
        # 返回的答案
        ans = 0

        # 开始遍历数组
        for i, num in enumerate(nums):
            # 如果数字是1，counter自增1
            if num:
                counter += 1
            # 如果是0，counter减掉1
            else:
                counter -= 1
            # 在前缀字典中查找当前位置的1和0个数的差值是否有相等的位置
            if counter in nums_hash:
                # 更新答案，因为两个位置的1和0的数量差值相等的充分必要条件是：这两个位置中间的1和0的个数相同
                ans = max(ans, i - nums_hash[counter])
            else:
                # 不存在相等的位置就放入字典中，用于后续的判断
                nums_hash[counter] = i
            
        return ans

参考文献：
https://leetcode-cn.com/problems/contiguous-array/solution/python-qian-zhui-he-bian-liang-zi-dian-b-9oct/