Напиши программу на питоне. Ваня хочет построить из кубиков n пирамидок. Он собирает пирамидки следующим образом: 1. Изначально в каждой пирамидке имеется по a1 кубиков в первом ряду. 2. В каждой второй пирамидке, т.е. номер которой делится на 2, к ним добавляется по a2 кубиков во второй ряд. 3. В каждой четвёртой пирамидке, т.е. номер которой делится на 4, добавляется по a3 кубиков в третий ряд. 4. В каждой восьмой пирамидке, т.е. номер которой делится на 8, добавляется по a4 кубиков в четвёртый ряд. И так далее. Выразим условие более формально: в каждой пирамидке, номер которой делится на (i−1)-ую степень числа 2, к исходному числу кубиков a1 добавляется по ai кубиков в i ряд. Помогите Ване определить количество кубиков, которое ему нужно для построения всех пирамидок

def total_cubes(n, a):
    total = 0
    for i in range(1, n+1):
        total += a
        a *= 2
    return total

n = int(input("Введите количество пирамидок: "))
a = int(input("Введите количество кубиков в первом ряду: "))

result = total_cubes(n, a)
print("Общее количество кубиков, необходимых для построения всех пирамидок:", result)

Пример работы программы:

Введите количество пирамидок: 5
Введите количество кубиков в первом ряду: 1
Общее количество кубиков, необходимых для построения всех пирамидок: 31