求斐波那契数列的第 n 项,n <= 39。
解题思路
如果使用递归求解,会重复计算一些子问题。例如,计算 f(4) 需要计算 f(3) 和 f(2),计算 f(3) 需要计算 f(2) 和 f(1),可以看到 f(2) 被重复计算了。
递归是将一个问题划分成多个子问题求解,动态规划也是如此,但是动态规划会把子问题的解缓存起来,从而避免重复求解子问题。
class Solution:
def __init__(self):
self.fib = list(range(40))
def Fibonacci(self, n):
# write code here
if n <= 1:
return n
for i in range(2, n + 1):
self.fib[i] = self.fib[i - 1] + self.fib[i - 2]
print(self.fib)
return self.fib[n]
class Solution:
def Fibonacci(self, n):
# write code here
res = [0, 1, 1, 2]
while len(res) <= n:
res.append(res[-1] + res[-2])
return res[n]
考虑到第 i 项只与第 i-1 和第 i-2 项有关,因此只需要存储前两项的值就能求解第 i 项,从而将空间复杂度由 O(N) 降低为 O(1)。
class Solution:
def Fibonacci(self, n):
# write code here
if n <= 1:
return n
pr1, pr2 = 0, 1
for i in range(2, n + 1):
res = pr1 + pr2
pr1, pr2 = pr2, res
return res