Задача к ЕГЭ по информатике на тему «Мусорки, кольцевая дорога» №3

Дано число N  – количество мусорок, расположенных по кольцу, затем ровно N  чисел – количество фантиков в каждой мусорке. Какова будет суммарная стоимость доставки фантиков, если расположить пункт сбора в середине?

Замечание. Стоимость доставки фантиков равно произведению количества фантиков на квадрат расстояния от сборщика до мусорки.

Входные данные

Дано два входных файла (файл A и файл B), каждый из которых в первой строке содержит число N (1 ≤ N ≤ 10000000)  – количество ресторанов. В каждой из следующих 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)

Ответ: 25434512 298289577519
Оцените статью
Я решу все!