Leetcode 1137:第 N 个泰波那契数(超详细的解法!!!)

泰波那契序列 T n T_n Tn定义如下:

$T_0 = 0, T_1 = 1, T_2 = 1, 且 在 ‘ n > = 0 ‘ 的 条 件 下 且在`n >= 0`的条件下 n>=0T_{n+3} = T_n + T_{n+1} + T_{n+2}$

给你整数 n,请返回第n个泰波那契数 T n T_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

解题思路

计算第 N 个泰波那契数非常简单,我这里使用python yield实现。

def tri(x):
    n, a, b, c = 0, 0, 1, 1
    while n < x:
        yield a + b + c
        a, b, c = b, c, a + b + c
        n += 1

因为n最大是32,所以我们可以先把所有的数给算出来,然后只要对输入的数在字典中查询即可。

class Solution:
    d = dict()
    def tribonacci(self, n: int) -> int:
        def tri(x):
            n, a, b, c = 0, 0, 1, 1
            while n < x:
                yield a + b + c
                a, b, c = b, c, a + b + c
                n += 1
                
        if n in self.d:
            return self.d[n]
        
        for i, v in enumerate(tri(35)):
            self.d[i + 3] = v
        self.d[0], self.d[1], self.d[2] = 0, 1, 1
        return self.d[n]

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页