Question

我有一个以这种方式定义的函数：

F(n) = n if n<=3

F(n) = F(n-1) + 2 * F(n-2) + 3 * F(n-3) if n>3

现在，我已将其编写为递归函数，并且工作正常。我试图把它写成一个迭代函数，但我似乎无法实现它。

输出应该是，例如：

print(FRec(5))  =>     22
print(FRec(10)) =>   1657
print(FRec(15)) => 124905

任何提示？

这是我的递归实现：

def FRec(n):
    if(n <= 3):
        return n
    if(n > 3):
        return FRec(n - 1) + 2 * FRec(n - 2) + 3 * FRec(n - 3)

Answer 1

您需要的只是保留最后3个结果：

from collections import deque

def F_iter(n):
    if n <= 3:
        return n
    prev = deque([1, 2, 3], maxlen=3)
    for i in range(4, n + 1):
        result = prev[-1] + 2 * prev[-2] + 3 * prev[-3]
        prev.append(result)
    return prev[-1]

如果deque不可用＆＃39;对你而言，你可以用一些列表切片和连接来低效地替换它：

    prev = [1, 2, 3]
    for i in range(4, n + 1):
        result = prev[-1] + 2 * prev[-2] + 3 * prev[-3]
        prev = prev[1:] + [result]

演示：

>>> F_iter(5)
22
>>> F_iter(10)
1657
>>> F_iter(15)
124905

这个递归函数的迭代版本

1 个答案: