Дано число – количество мусорок, расположенных по кольцу, затем ровно
чисел – количество фантиков в каждой мусорке. Какова будет суммарная стоимость доставки фантиков, если расположить пункт сбора в середине?
Замечание. Стоимость доставки фантиков равно произведению количества фантиков на квадрат расстояния от сборщика до мусорки.
Входные данные
Дано два входных файла (файл A и файл B), каждый из которых в первой строке содержит число N – количество ресторанов. В каждой из следующих N строк находится одно число: количество фантиков в каждой мусорке, все числа натуральные. Мусорки перечислены в порядке их расположения вдоль дороги, начиная от нулевой отметки.
В ответе укажите два числа: сначала значение искомой величины для файла А, затем – для файла B.
f = open(’27B.txt’)
n = int(f.readline())
a = [int(f.readline()) for _ in range(n)]
cost = 0
middle = n // 2 # Находим индекс середины
for i in range(n):
distance_to_middle = abs(i - middle) # Расстояние от текущей мусорки до середины
cost += a[i] * (distance_to_middle ** 2)
print(cost)