欧拉计划 14

以下迭代序列定义在整数集合上：

n → n/2 (当n是偶数时)
n → 3n + 1 (当n是奇数时)

应用以上规则，并且以数字13开始，我们得到以下序列：

13 → 40 → 20 → 10 → 5 → 16 → 8 → 4 → 2 → 1
可以看出这个以13开始以1结束的序列包含10个项。
虽然还没有被证明（Collatz问题），但是人们认为在这个规则下，以任何数字开始都会以1结束。

以哪个不超过100万的数字开始，能给得到最长的序列？
注意: 一旦序列开始之后，也就是从第二项开始，项是可以超过100万的。

def get_collatz_len(n):
    """ 
    获取序列长度
    保存各个数字开始的序列，求取新的序列时，从已获知的序列字典中查询
    """
    collatz_list = []
    while 1:
        if n in collatz_len_dict:
            for i, collatz in enumerate(collatz_list):
                collatz_len_dict[collatz] = len(collatz_list) - i + collatz_len_dict[n]
            return collatz_len_dict[n] + len(collatz_list)
        collatz_list.append(n)
        if n % 2:
            n = 3 * n + 1
        else:
            n //= 2


#def get_len(n):
#    """ 获取序列长度 """
#    collatz_len = 1
#    while n != 1:
#        if n % 2:
#            n = 3 * n + 1
#        else:
#            n /= 2
#        collatz_len += 1
#    return collatz_len


# 保存各个数字开始的序列
collatz_len_dict = {1: 1}
len_max = 0
i_collatz = 0
for i in range(1, 10 ** 6):
    i_len = get_collatz_len(i)
    if i_len > len_max:
        len_max = i_len
        i_collatz = i
print(i_collatz)