【leetcode】525. Contiguous Array

最新推荐文章于 2025-08-20 21:49:59 发布

weixin_33670713

最新推荐文章于 2025-08-20 21:49:59 发布

阅读量51

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/seyjs/p/8989287.html

本文介绍了一种通过遍历数组并使用哈希表记录0和1数量差值的方法，来寻找包含相等数量0和1的最长子数组的有效算法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目如下：

解题思路：这个题目可以这么做，遍历数组，如果元素是0，则count --；否则count ++；这样的话，每遍历到一个下标i，count的值就是0>i区间内0和1的差值。如果我们能找到在0->i-1区间内差值也为count的最小下标j，那么j->i区间就是一个0和1数量相同的区间。遍历过程中记录最大的j->i长度即可得到答案。

代码如下：

class Solution(object):
    def findMaxLength(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        dic = {}
        count = 0
        res = 0
        for i,v in enumerate(nums):
            if v == 0:
                count -= 1
            else:
                count += 1
            if count == 0: # count = 0 表示0->i 区间内0和1数量相同，肯定是当前的最大长度
                res = i+1
                continue
            if dic.has_key(count): 
                res = max(res,i - dic[count]) #找到历史出现的count的最小下标，差值即为一个符合条件的区间的长度
            if dic.has_key(count) == False: #记录每个count出现的最小下标
                dic[count] = i
        #print dic
        return res