最长的Collat​​z序列 - Memoization- Python-迭代与递归

时间:2016-02-17 16:44:32

标签: python-3.x dynamic-programming memoization

dict={}
def recur(n):
    counter=1
    original=n
    while n!=1:
        n = n/2 if n % 2 == 0 else 3*n+1
        if n in dict:
            counter=counter+dict[n]
            dict[original]=counter
            return counter

        counter=counter+1
    dict[original]=counter
    return counter
for i in range(1,1000000):
    recur(i)
print(max(dict.keys(), key=(lambda k: dict[k])))

如何记忆一次通话中使用的所有号码?例如,当我调用recur(13)时,它只会在dict中存储13的值,而不会在recur中使用40,20,10,5等值(13)

另外,我无法生成递归函数,因为我可以计算(通过在函数中添加计数器参数)但是我无法在字典中添加值。

请建议一种方法,以便将最大可能值存储在内存中,并且该函数是递归的吗?

1 个答案:

答案 0 :(得分:1)

这是编写我能想到的递归函数的最可读(也是相当pythonic)的方法;它基本上只是阐明了建立序列的规则,因为你要向某人解释:

count = {}

def recur(n):
    if n not in count:  # memoize
        if n == 1:
            count[n] = 0  # finished
        else:
            count[n] = recur(n//2 if n % 2 == 0 else 3*n + 1)
        count[n] += 1  # this step
    return count[n]

for i in range(1, 100):
    recur(i)

for item in sorted(count.items()):
    print(item)

使用count初始化1缓存可以进行优化,但会牺牲将编队规则直接转换为代码:

count = {1: 1}

def recur(n):
    if n not in count:  # memoize
        count[n] = 1 + recur(n//2 if n % 2 == 0 else 3*n + 1)
    return count[n]