525. 连续数组(中等)(LCR 011)

https://leetcode.cn/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的最长连续子数组。

class Solution:
    def findMaxLength(self, nums: List[int]) -> int:
        # 数组、哈希表、前缀和
        # 0和1的数量相同，等价于1的数量减去0的数量等于0
        # 我们可将数组中0视作-1，则原问题转换为求最长的连续子数组，其元素和为0
        presum, maxlength = 0, 0
        # 定义哈希表，键值对为 前缀和：首次出现该前缀和时的索引
        # 规定空的前缀和的下标为-1，空前缀和的元素和为0，因此在遍历前在哈希表中存入键值对(0, -1)
        dic = {0: -1}
        for k, v in enumerate(nums):
            # 遍历数组，遇到1时，presum加1，遇到0时，presum减1
            if v == 1:
                presum += 1
            else:
                presum -= 1
            # 关键：如果当前前缀和出现在字典中，说明该前缀和在前面出现过，不断添加n个元素后，前缀和仍为
            # 当前前缀和的值，说明中间的这n个元素和为0。
            if presum in dic:
                # 此时更新最终结果，用当前下标减去第一次出现的下标。即为当前最长连续子数组
                # 此时无需更新字典中的下标位置，是因为如果后面再次出现相同的前缀和，仍说明
                # 这中间的m个元素和为0。
                firstindex = dic[presum]
                # 之所以空前缀下标是-1，是因为求长度，比如k=2，从0到2是3个数，此时默认-1，2-(-1) = 3
                maxlength = max(maxlength, k - firstindex)
            else:
                # 若没出现则将当前第一次出现的前缀和添加到字典中。
                dic[presum] = k
        return maxlength