原题
大家都知道斐波那契数列,现在要求输入一个整数n,请你输出斐波那契数列的第n项(从0开始,第0项为0)。
n<=39
Reference Answer
解题思路:
直接用回溯法了,用递归效率低。
# -*- coding:utf-8 -*-
class Solution:
def Fibonacci(self, n):
# write code here
res = []
res.append(0)
res.append(1)
# res[0], res[1] = 0, 1
if n > 1:
for i in range(2, n+1):
res.append(res[i-1] + res[i-2])
return res[n]
这种方法得到时间复杂度是O(n),空间复杂度也是O(n);
优化版:
可进一步压缩空间复杂度,这样在一次查询的时候进一步降低空间复杂度为O(1);
C++ Version:
class Solution {
public:
int Fibonacci(int n) {
int base0 = 0, base1 = 1;
if (n == 0){
return base0;
}
else if (n == 1){
return base1;
}
int res = 0;
for(int i = 2; i <= n; ++i){
res = base0 + base1;
base0 = base1;
base1 = res;
}
return res;
}
};
即不再保存中间结果,进一步减低了空间复杂度。