Skip to content

Commit 02d89bd

Browse files
Added implementation for Tribonacci sequence using dp (TheAlgorithms#6356)
* Added implementation for Tribonacci sequence using dp * Updated parameter name * Apply suggestions from code review --------- Co-authored-by: Tianyi Zheng <[email protected]>
1 parent f24ab2c commit 02d89bd

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

dynamic_programming/tribonacci.py

+24
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
# Tribonacci sequence using Dynamic Programming
2+
3+
4+
def tribonacci(num: int) -> list[int]:
5+
"""
6+
Given a number, return first n Tribonacci Numbers.
7+
>>> tribonacci(5)
8+
[0, 0, 1, 1, 2]
9+
>>> tribonacci(8)
10+
[0, 0, 1, 1, 2, 4, 7, 13]
11+
"""
12+
dp = [0] * num
13+
dp[2] = 1
14+
15+
for i in range(3, num):
16+
dp[i] = dp[i - 1] + dp[i - 2] + dp[i - 3]
17+
18+
return dp
19+
20+
21+
if __name__ == "__main__":
22+
import doctest
23+
24+
doctest.testmod()

0 commit comments

Comments
 (0)