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)}")