0525：连续数组（★）

力扣第 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

分析

假如将 0 都替换为 -1，那么就是找和为 0 的子数组。

求任意子数组的和，容易想到前缀和。得到前缀和数组 pre 后，就是求距离最远的相同数，用哈希表即可。

解答

1
2
3

def findMaxLength(self, nums: List[int]) -> int:
    d, pre = {}, accumulate([0]+nums, lambda x, y: x+(y if y==1 else -1))
    return max(i-d.setdefault(val, i) for i, val in enumerate(pre))

212 ms