leetcode-1137. 第 N 个泰波那契数

81 阅读1分钟

题目要求:

泰波那契序列 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.cn/problems/n-…

解题思路

属于是动态规划的简单题目,思路比较简单。

代码

func tribonacci(n int) int {
	// 使用动态规划
	if n == 0 {
		return 0
	}
	if n <= 2 {
		return 1
	}
	a, b, c, d := 0, 1, 1, 2
	for i := 3; i < n; i++ {
		a = b
		b = c
		c = d
		d = a + b + c
	}
	return d
}