Відповідь:
a=int(input("Введіть кількість ярусів:n"))
t=0
for i in range(1,a+1):
t+=i*2
print(f"Для поливу дерева з кількістю {a} ярус/ярусів/яруси потрібно {t+1} літрів води")