题目来自LeetCode,链接:面试题10- I. 斐波那契数列。具体描述:写一个函数,输入 n ,求斐波那契(Fibonacci)数列的第 n 项。斐波那契数列的定义如下:
F(0) = 0, F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 N > 1.
斐波那契数列由 0 和 1 开始,之后的斐波那契数就是由之前的两数相加而得出。答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例1:
输入:n = 2
输出:1
示例2:
输入:n = 5
输出:5
这题就很简单,相信大家学递归/递推的时候都学过这个经典的例子,就不再赘述了,需要注意的一点是取模,不是在结果取模,而是中间的运算过程就需要取模,防止溢出。这里用的递推方法,时间复杂度为 O ( n ) O(n) O(n),空间复杂度为 O ( 1 ) O(1) O(1)。
JAVA版代码如下:
class Solution {
public int fib(int n) {
if (n == 0 || n == 1) {
return n;
}
int F0 = 0, F1 = 1;
for (int i = 2; i <= n; ++i) {
int temp = F1;
F1 = (F0 + F1) % 1000_000_007;
F0 = temp;
}
return F1;
}
}
提交结果如下:
Python版代码如下:
class CQueue:
def __init__(self):
self.stack_in = []
self.stack_out = []
def appendTail(self, value: int) -> None:
self.stack_in.append(value)
def deleteHead(self) -> int:
if len(self.stack_in) == 0 and len(self.stack_out) == 0:
return -1
if len(self.stack_out) == 0:
while len(self.stack_in) > 0:
self.stack_out.append(self.stack_in.pop())
return self.stack_out.pop()
# Your CQueue object will be instantiated and called as such:
# obj = CQueue()
# obj.appendTail(value)
# param_2 = obj.deleteHead()
提交结果如下: