连续数组（LeetCode）

好好学习Py

于 2024-08-14 21:00:04 发布

阅读量131

点赞数 3

分类专栏：算法与数据结构文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/weixin_74254879/article/details/141200287

版权

算法与数据结构专栏收录该内容

40 篇文章 0 订阅

订阅专栏

题目

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

解题

"""
时间复杂度：O(n)，因为我们只遍历数组一次。
空间复杂度：O(n)，因为我们需要存储前缀和的哈希表。
"""


def findMaxLength(nums):
    prefix_sum_map = {0: -1}  # 初始化前缀和为 0 的位置为 -1，便于计算长度
    max_length = 0
    prefix_sum = 0

    for i, num in enumerate(nums):
        # 将 0 转换为 -1
        if num == 0:
            prefix_sum -= 1
        else:
            prefix_sum += 1

        if prefix_sum in prefix_sum_map:
            # 当前前缀和已存在，计算子数组长度
            max_length = max(max_length, i - prefix_sum_map[prefix_sum])
        else:
            # 当前前缀和第一次出现，记录其位置
            prefix_sum_map[prefix_sum] = i

    return max_length


nums = [0, 1, 0]
print(findMaxLength(nums))  # 输出: 2