如何改进代码以避免基于内存的错误?

时间:2017-05-24 00:49:20

标签: python out-of-memory

我编写了一个python函数largestProduct(n),它返回由两个n位数字的乘积组成的最大数字。该代码适用于n直到3,但显示n>3的内存错误。有没有办法可以改进我的代码以避免这个错误?

def largestProduct(n):

    number_lst = []
    prod_lst = []
    count = 0
    for i in range(10**(n-1),(10**n)):
        number_lst.append(i) 
    while count!= len(number_lst):
        for i in range(len(number_lst)):
            prod_lst.append(number_lst[count]*number_lst[i])
        count +=1
    prod_set = list(set(prod_lst))

    return max(prod_lst)

2 个答案:

答案 0 :(得分:3)

嗯,你不需要任何存储来循环你需要的东西:

def largestProduct(n):
    range_low = 10**(n-1)
    range_high = 10**n
    largest = 0
    # replace xrange with range for Python 3.x
    for i in xrange(range_low, range_high):
        for j in xrange(range_low, range_high):
            largest = max(largest, i*j)
    return largest

但你为什么要这样做呢?两个 n - 长数的最大乘积总是用 n 位数表示的最大数字,即:

def largestProduct(n):
    return (10**n-1)**2

答案 1 :(得分:2)

您应该考虑创建一个生成器函数。最后,您可以遍历函数的输出,该输出仅逐个处理每个元素,而不是将整个列表保存在内存中。

请参阅https://wiki.python.org/moin/Generators

相关问题