写一个函数,输入 n
,求斐波那契(Fibonacci)数列的第 n
项(即 F(N)
)。斐波那契数列的定义以下:golang
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:code
输入:n = 5 输出:5
golang递归
//当递归的栈数过多,重复计算的子过程过长,产生超时 func fib(n int) int { if n==0 { return 0 } if n==1 { return 1 } return ((fib(n-1)+fib(n-2))%1000000007) }
采用
动态规划
的方法:
动态规划三部曲:
1.初始状态:dp[0]=0,dp[1]=1,其中dp[i]表示斐波那契数列的第i个数
2.状态转移方程:dp[n]=dp[n-1]+dp[n-2],即对应的f(n)=f(n-1)+f(n-2)
3.返回值:dp[n],对应第n个斐波那契数列的值ci
func fib(n int) int{ if n == 0 || n == 1{ return n } dp := make([]int,n+1) dp[0],dp[1] = 0,1 //初始状态 for i:=2;i<=n;i++{ dp[i] = (dp[i-1]+dp[i-2])%1000000007 //状态转移方程 } return dp[n] //返回值 }
空间优化:由于斐波那契数列主要就是三个数,f(n),f(n-1)和f(n-2),而上面的动态规划采用了一个数组存储斐波那契数列,空间复杂度为O(n),这里采用三个变量sum,a,b分别表示f(n),f(n-1)和f(n-2),将空间复杂度下降到O(1)leetcode
func fib(n int) int { a, b, sum := 0, 1, 0 for i := 0; i < n; i++ { sum = (a + b) % 1000000007 a, b = b, sum } return a }
/** * @param {number} n * @return {number} */ var fib = function(n) { let a=0,b=1,sum=0; for(let i=0;i<n;i++){ sum = (a+b)%1000000007; [a,b] = [b,sum]; } return a; };