斐波那契数动态规划算法

Problem: 509. 斐波那契数

思路

斐波那契数的边界条件是 F(0)=0 和 F(1)=1。当 n>1 时，每一项的和都等于前两项的和，因此有如下递推关系：
F(n)=F(n−1)+F(n−2)

解题方法

因此可以使用动态规划求解。动态规划的状态转移方程即为上述递推关系，边界条件为 F(0) 和 F(1)。

根据状态转移方程和边界条件，可以得到时间复杂度和空间复杂度都是 O(n) 的实现。由于F(n) 只和 F(n−1) 与 F(n−2) 有关，因此可以使用「滚动数组思想」把空间复杂度优化成 O(1)。如下的代码中给出的就是这种实现。

复杂度

• 时间复杂度: O(n)

• 空间复杂度: O(1)

Code

class Solution {
    func fib(_ n: Int) -> Int {
        var f0 = 0, f1 = 1
        if n == 0 {
            return f0
        } else if n == 1 {
            return f1
        }

        var temp = 0
        for i in 2...n {
            temp = f0 + f1
            f0 = f1
            f1 = temp
        }
        return temp
    }
}