[LeetCode] 1137. 第N个泰波那契数

福楠2024-11-07 14:01

题目描述：

泰波那契序列 Tn 定义如下：

T0 = 0, T1 = 1, T2 = 1, 且在 n >= 0 的条件下 Tn+3 = Tn + Tn+1 + Tn+2

给你整数 n，请返回第 n 个泰波那契数 Tn 的值。

示例 1：

复制代码

输入：n = 4
输出：4
解释：
T_3 = 0 + 1 + 1 = 2
T_4 = 1 + 1 + 2 = 4

示例 2：

复制代码

输入：n = 25
输出：1389537

提示：

0 <= n <= 37
答案保证是一个 32 位整数，即 answer <= 2^31 - 1。

题目链接：

. - 力扣（LeetCode）

解题主要思路：

妥妥的动态规划，需要注意的是泰波那契数是从0开始，因此第n个泰波那契数，我们需要开的空间是n+1个。

不过，做完后发现，我们始终是只需要四个int变量，因此我们可以采用滚动数组的方式来进行空间优化。

解题代码：

（非优化版本）

cpp 复制代码

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

（空间优化 -- 滚动数组版本）

cpp 复制代码

class Solution {
public:
    int tribonacci(int n) {
        if (n == 0) return 0;
        if (n == 1 || n == 2) return 1;
        int a, b, c, dst;  // 四个int类型变量来代替dp表
        a = 0, b = c = 1;  // 初始化
        for (int i = 3; i < n+1; ++i) {
            dst = a + b + c;
            a = b, b = c, c = dst;
        }
        return dst;
    }
};