На вход программы поступает последовательность из N целых положительных чисел. Рассматриваются тройки различных элементов последовательности, в которых между любыми двумя элементами расстояние не менее 99 (расстояние – разность порядковых номеров). Необходимо определить максимальную сумму такой тройки, кратную 123. Причём каждый элемент тройки должен делиться на 3 без остатка.
Входные данные: Даны два входных файла, каждый из которых содержит в первой строке количество пар N . Каждая из следующих N строк содержит одно натуральное число.
f = open(’10_27B_pairs.txt’)
n = int(f.readline())
a = [int(i) for i in f]
d = 99
k = 123
# Список из максимальных чисел с определенным остатком от деления
t = [-1] * k
# Список из пар с максимальными суммами c определенным остатком
v = [[-1, -1] for i in range(k)]
mx = -1
for i in range(d * 2, len(a)):
if a[i - d * 2] % 3 == 0:
# Обрабатываем элемент на расстоянии 2*d от текущего
ost1 = a[i - d * 2] % k
# Если он больше прошлого с таким остатком - обновляем список
t[ost1] = max(t[ost1], a[i - d * 2])
if a[i - d] % 3 == 0:
# Обрабатываем средний элемент - на расстоянии d от текущего
ost2 = a[i - d] % k
# Для каждого из максимальных первых элементов (для каждого остатка)
# создаём новые суммы с новым средним элементом и обновляем
# эти суммы в списке v, если они получились больше прошлых
for j in range(k):
if t[j] > -1:
sm_pair = t[j] + a[i - d]
if sm_pair > sum(v[(j + ost2) % k]):
v[(j + ost2) % k] = [t[j], a[i - d]]
if a[i] % 3 == 0:
# Вычисляем остаток для пары в сумму к нашему числу
ost3 = (k - (a[i] % k)) % k
# Если уже нашлось число с таким остатком - считаем и обновляем максимум
if sum(v[ost3]) > -1:
sm = a[i] + sum(v[ost3])
if sm > mx:
mx = sm
print(mx)
Ответ: 2912886 2999970