力扣1137. 第 N 个泰波那契数

最新推荐文章于 2024-10-08 12:37:10 发布

我是一颗什么糖

最新推荐文章于 2024-10-08 12:37:10 发布

阅读量105

点赞数

分类专栏：力扣刷题文章标签： python 动态规划

本文链接：https://blog.csdn.net/qq_41596063/article/details/119522815

版权

力扣刷题专栏收录该内容

58 篇文章 0 订阅

订阅专栏

本文介绍了如何使用O(n)时间复杂度和O(1)空间复杂度的自顶向下方法求解第n个泰波那契数Tn。通过迭代更新前三个数的值，避免了存储大量中间结果。适合解决LeetCode问题Nth Tribonacci Number。

摘要由CSDN通过智能技术生成

题目描述

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

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

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

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

示例

示例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。

思路&代码

第一眼看这道题，立马想到斐波那契数列，解法都类似。下面将思路和代码进行展示。（其实也可以用递归做，但会超时，故不介绍）

思路一：自底向上计算，时间复杂度O(n)，空间复杂度O(n)。

class Solution:
    def tribonacci(self, n: int) -> int:
        cache = {0: 0, 1: 1, 2: 1}
        if n in cache: 
            return cache[n]
        for cursor in range(2, n):
            cache[cursor+1] = cache[cursor] + cache[cursor-1] + cache[cursor-2]
        return cache[n]

思路二：自顶向下计算，时间复杂度O(n)，空间复杂度O(1)。

class Solution:
    def tribonacci(self, n: int) -> int:
        if n == 0: return 0
        front, mid, rear = 0, 1, 1
        while n >= 3:
            front, mid, rear = mid, rear, front+mid+rear
            n -= 1
        return rear