tree.py

def sum_tree(w, h):
  summ = 0
  for i in range(0, h + 1):
    print(f"{summ} + {w**i}")
    summ += w**i
    
  return summ

def formula_tree(w, h):
  return (w**(h + 1) - 1) // (w - 1)

w = 3
h = 4
print(f"Sum: {sum_tree(w, h)}")
print(f"Formula: {formula_tree(w, h)}")