В текстовом файле записан набор натуральных чисел, не превышающих . Гарантируется, что все числа различны. Из набора нужно выбрать три числа, сумма которых делится на
. Какую наибольшую сумму можно получить?
Первая строка входного файла содержит натуральное число — общее количество чисел в файле. Каждая из следующих
строк содержит одно число.
Пример входного файла:
Для данного примера в ответе надо записать число .
Вам даны два входных файла, каждый из которых имеет описанную выше структуру. В ответе укажите два числа, ответ для файла и для файла
.
Решение 1. Неэффективное
f = open("27-A.txt")
n = int(f.readline())
a = []
for i in range(n):
a.append(int(f.readline()))
k = 0
for i in range(n):
for j in range(i + 1, n):
for l in range(j + 1, n):
if (a[i] + a[j] + a[l]) % 10 == 0:
k = max(k, a[i] + a[j] + a[l])
print(k)
Решение 2. Эффективное. Динамика
f = open(’16.txt’)
n = int(f.readline())
a = [int(i) for i in f]
k = 10
# Список из максимальных чисел с определенным остатком от деления
t = [-1] * k
# Список из пар с максимальными суммами с определенным остатком
v = [[-1, -1] for i in range(k)]
mx = -1
for i in range(2, len(a)):
# Обрабатываем элемент на расстоянии 2 от текущего (через один слева)
ost1 = a[i - 2] % k
# Если он больше прошлого с таким остатком - обновляем список
t[ost1] = max(t[ost1], a[i - 2])
# Обрабатываем средний элемент - на расстоянии 1 от текущего (слева)
ost2 = a[i - 1] % k
# Для каждого из максимальных первых элементов (для каждого остатка)
# создаём новые суммы с новым средним элементом и обновляем
# эти суммы в списке v, если они получились больше прошлых
for j in range(k):
if t[j] > -1:
sm_pair = t[j] + a[i - 1]
if sm_pair > sum(v[(j + ost2) % k]):
v[(j + ost2) % k] = [t[j], a[i - 1]]
# Вычисляем остаток для пары в сумму к нашему числу
ost3 = (k - (a[i] % k)) % k
# Если уже нашлось число с таким остатком - считаем и обновляем максимум
if sum(v[ost3]) > -1:
sm = a[i] + sum(v[ost3])
if sm > mx:
mx = sm
print(mx)