Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
Recursion with memoization solution in Clear category for Golden Pyramid by r_tchaik
from functools import lru_cache
@lru_cache
def count_gold(pyramid, row=0, col=0):
return pyramid[row][col] + max(count_gold(pyramid, row + 1, col + idx) for idx in (0, 1)) \
if row < len(pyramid) else 0
if __name__ == '__main__':
#These "asserts" using only for self-checking and not necessary for auto-testing
assert count_gold((
(1,),
(2, 3),
(3, 3, 1),
(3, 1, 5, 4),
(3, 1, 3, 1, 3),
(2, 2, 2, 2, 2, 2),
(5, 6, 4, 5, 6, 4, 3)
)) == 23, "First example"
assert count_gold((
(1,),
(2, 1),
(1, 2, 1),
(1, 2, 1, 1),
(1, 2, 1, 1, 1),
(1, 2, 1, 1, 1, 1),
(1, 2, 1, 1, 1, 1, 9)
)) == 15, "Second example"
assert count_gold((
(9,),
(2, 2),
(3, 3, 3),
(4, 4, 4, 4)
)) == 18, "Third example"
Dec. 2, 2020
