从一道算法题看用空间换时间的思维方式

最近刷到了一道很好的算法题:

写一个函数,输入 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。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/fei-bo-na-qi-shu-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

于是乎,这还不简单,我很轻松的写下了如下的代码:

class Solution {
    public int fib(int n) {
        if(n < 2){
            return n;
        }
        return (fib(n-1) + fib(n-2))%1000000007;
    }
}

 Leetcode提示运行超时,我们举个栗子,以f(6)为例:

很明显,我们进行了大量的重复运算,使得时间复杂度大大增加!

 优化:

class Solution {
    public int fib(int n) {
        if(n < 2){
            return n;
        }
        int [] dp = new int[n+1];
        dp[0] = 0;
        dp[1] = 1;
        for(int i = 2;i <= n;i++){
            dp[i] = dp[i-1] + dp[i-2];
            dp[i] %= 1000000007;
        }
        return dp[n];
    }
}

我们定义一个数组,用空间存储f(n)的值,这样避免了大量的重复计算,大大降低了时间复杂度!

猜你喜欢

转载自blog.csdn.net/qq_36428821/article/details/112299362
今日推荐