【LeetCode】第 N 个泰波那契数 (认识动态规划)

链接: 第 N 个泰波那契数

编写代码

class Solution {
public:
    int tribonacci(int n) {
        if(n == 0)
        {
            return 0;
        }
        else
        {
            if(n ==1 || n == 2)
            return 1;
        }
        vector<int> dp(n + 1);
        dp[0] = 0;dp[1] = 1;dp[2] = 1;
        for(int i = 3;i <= n;i++)
        {
            dp[i] =dp[i-3] + dp[i -2] + dp[i - 1];
        }

        return dp[n];
    }
};

代码空间优化

一般像这种情况我们可以使用滚动数组的方式来解决空间的问题

class Solution {
public:
    int tribonacci(int n) {
        if(n == 0)
        {
            return 0;
        }
        else
        {
            if(n ==1 || n == 2)
            return 1;
        }
        int a = 0,b = 1 , c = 1, d = 0;
        for(int i = 3;i <= n;i++)
        {
            d = a + b + c;
            a = b;
            b = c;
            c = d;
        }

        return c;
    }
};