Как оптимизировать этот код: ввод Leetcode был неожиданным (превышен лимит памяти) ⇐ Python

[Anonymous]

Вопрос по Leetcode: учитывая числа двоичного массива, верните максимальную длину непрерывного подмассива с равным количеством 0 и 1.
Leetcode ввел в функцию очень длинный ввод и его IDE не могла справиться с завышенным использованием памяти
(обновление: добавлены 2 лучшие версии)
В первые версии я добавил все возможное подмассивы в 2d список. Я удалил это в этой версии. Теперь я обрабатываю каждый список после его создания.
Leetcode по-прежнему заявляет, что время превышено.
это часть их ввода (он вдвое или втрое превышает эту длину). )
[1,1,0,1,0,1,0,0,0,0,0,0,0,0,1,0,0,1,0,0,1,1,1,0,1,1,1,1,1,1,1,1,0,1,0,0,1,0,0,1,1,1,1,0,0,1,0,0,0,1,1,1,1,0,0,0,1,0,0,0,0,1,0,0,1,0,1,1,1,0,0,0,1,1,0,0,1,1,0,1,0,0,0,0,0,0,0,1,0,0,1,1,0,1,0,1,1,1,1,0,1,1,0,0,0,1,1,0,0,1,1,1,1,0,0,0,1,1,1,0,1,1,0,1,0,0,1,0,1,1,1,0,0,1,0,1,1,0,0,0,1,1,0,0,1,0,1,1,1,1,1,1,0,1,0,0,1,0,0,1,1,1,1,1,0,1,1,1,1,1,0,0,1,1,0,1,1,0,0,0,1,1,1,1,

Будем благодарны за помощь в оптимизации:
class Solution(object):

#version 2

def findMaxLength2( nums):
"""
:type nums: List[int]
:rtype: int
"""
l_sub = [] #list of subarray derived from nums
LEN = len(nums)
_max = 0
len_max = 0

# divides into subarrays
for i in range(LEN):
count = [0] * 2
for j in range(i + 1, LEN + 1):
_str = "".join([str(_m) for _m in nums[i:j]])
count[0]= _str.count("0")
count[1] = _str.count ("1")
if count[0] == count[1] and count[1] >= _max :
_max = count[0]
len_max = _max * 2
return len_max

#version 3:

def findMaxLength3( nums):
"""
:type nums: List[int]
:rtype: int
"""
l_sub = [] #list of subarray derived from nums
LEN = len(nums)
_max = 0
len_max = 0

# divides into subarrays
for i in range(LEN):
for j in range(i + 1, LEN + 1):
ones = sum(filter(lambda x:x ==1 , nums[i:j]))
zeros= len (nums[i:j]) - ones
if ones == zeros and zeros >= _max :
_max = zeros
len_max = _max * 2

return len_max



Подробнее здесь: https://stackoverflow.com/questions/791 ... mit-exceed