1137. 第 N 个泰波那契数

题目

题目来源

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-th-tribonacci-number

题目描述

泰波那契序列 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。

解题

解题思路

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

T(n)=T(n-1)+T(n-2)+T(n-3)

由于泰波那契数存在递推关系，因此可以使用动态规划求解。动态规划的状态转移方程即为上述递推关系，边界条件为 T(0)、T(1) 和 T(2)。

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

代码

class Solution {
    public int tribonacci(int n) {
        int[] sun = new int[n + 1];
        for(int i = 0 ;i <= n ; i ++){
            if( i == 0){
                sun[i] = 0;
            }else if( i == 1){
                sun[i] = 1;
            }else if(i == 2){
                sun[i] = 1;
            }else{
                sun[i] = sun[i - 3] + sun[i - 2] + sun[i - 1];
            }
        }
        return sun[n];
    }
}