题目表述 泰波那契序列 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。
第一种方法:迭代求解
def tribonacci(self, n): if n <= 0: return 0 if n == 1 or n == 2: return 1 numa = 0 numb = 1 numc = 1 sum1 = 0 for i in range(3,n+1): sum1 = numa + numb + numc numa = numb numb = numc numc = sum1 return sum1
第二种方法:递归求解
#需要注意看给的测试用例,第N项是前三项之和,不是前两项
def tribonacci(n):
if n <= 0:
return 0
if n == 1 or n == 2:
return 1
return tribonacci(n - 1) + tribonacci(n - 2) + tribonacci(n - 3)
#input接收的参数是字符型
print(tribonacci(int(input())))