剑指 Offer II 011. 0 和 1 个数相同的子数组

最新推荐文章于 2022-05-05 14:40:05 发布

万物琴弦光锥之外

最新推荐文章于 2022-05-05 14:40:05 发布

阅读量9.1k

点赞数

文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/weixin_43702920/article/details/120288532

版权

本文介绍了如何使用Python实现一个解决方案，通过将0转换为-1并利用前缀和结合哈希表来找到具有相同数量0和1的最长连续子数组。方法涉及遍历数组、更新前缀和以及维护一个字典来记录每个前缀和出现的位置。

摘要由CSDN通过智能技术生成

题目

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

解法

0换成-1
前缀和 + 哈希表（字典）

class Solution:
    def findMaxLength(self, nums: List[int]) -> int:
        # preprocessing
        n = len(nums)
        preSum = [0] * (n+1)
        t = 0
        for i in range(n):
            if not nums[i]:
                nums[i] = -1
            t += nums[i]
            preSum[i+1] = t
            
        # print(preSum)

        # main
        dic = {}
        length = 0
        for i in range(n+1):
            if preSum[i] in dic:
                length = max(i - dic[preSum[i]], length)
                # print(i, length, dic)
                dic[ preSum[i] ] = min(dic[ preSum[i] ], i)
            else:
                dic[preSum[i]] = i
        return length