Python/dynamic_programming/factorial.py

39 lines
836 B
Python
Raw Normal View History

2019-10-05 05:14:13 +00:00
# Factorial of a number using memoization
result = [-1] * 10
result[0] = result[1] = 1
def factorial(num):
"""
>>> factorial(7)
5040
>>> factorial(-1)
'Number should not be negative.'
>>> [factorial(i) for i in range(5)]
[1, 1, 2, 6, 24]
"""
2019-10-05 05:14:13 +00:00
if num < 0:
return "Number should not be negative."
2019-10-05 05:14:13 +00:00
if result[num] != -1:
return result[num]
else:
2019-10-05 05:14:13 +00:00
result[num] = num * factorial(num - 1)
# uncomment the following to see how recalculations are avoided
# print(result)
return result[num]
2019-10-05 05:14:13 +00:00
# factorial of num
# uncomment the following to see how recalculations are avoided
# result=[-1]*10
# result[0]=result[1]=1
# print(factorial(5))
# print(factorial(3))
# print(factorial(7))
if __name__ == "__main__":
import doctest
2019-10-05 05:14:13 +00:00
doctest.testmod()