show_recursion_tree

prints out recursion tree

Installation

pip install show_recursion_tree

Example 1

@show_recursion_tree
def factorial(n):
    if n == 1:
        return 1
    return n * factorial(n-1)

print(factorial(5))

"""
  5:120 
    |
  4:24
   |
 3:6
  |
 2:2
  |
1:1

120
"""

Example 2

from show_recursion_tree import show_recursion_tree

@show_recursion_tree
def fib(n):
    if n == 1:
        return 1
    if n == 2:
        return 1
    
    return fib(n-2) + fib(n-1)

print(fib(5))
"""
      _________5:5__________
      |                    |
  ___3:2_____         ___4:3________
  |         |         |            |
1:1       2:1       2:1        ___3:2_____
                               |         |
                             1:1       2:1

5
"""

Example 3

@show_recursion_tree
def f(n):
    if n <= 4:
        return 1
    
    return f(n-4) + f(n-3) + f(n-2) + f(n-1)

f(7)

"""
  ___________________7:13______________________
  |    |             |                        |
3:1  4:1    ________5:4_________    ________6:7________________
            |    |        |    |    |    |      |             |
          1:1  2:1      3:1  4:1  2:1  3:1    4:1    ________5:4_________
                                                     |    |        |    |
                                                   1:1  2:1      3:1  4:1
"""