【leetcode】525. Contiguous Array

题目如下：

解题思路：这个题目可以这么做，遍历数组，如果元素是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