Задание выполняется с использованием прилагаемых файлов
В текстовом файле записан набор натуральных чисел, не превышающих Из набора нужно выбрать три числа, сумма которых делится на
. Какую наименьшую сумму можно при этом получить?
Пример входного файла
Первая строка входного файла содержит целое число – общее количество чисел в наборе. Каждая из следующих
строк содержит одно число.
Пример организации исходных данных во входном файле:
4
9
16
20
15
В данном случае есть две подходящие тройки: 9, 16, 15 (сумма 40) и 9, 20, 15 (сумма 44). В ответе надо записать число 40.
Вам даны два входных файла (A и B), каждый из которых имеет описанную выше структуру. В ответе укажите два числа: сначала значение искомой суммы для файла A, затем для файла B.
Решение 1 (неэффективное)
f = open("27A.txt")
n = int(f.readline())
a = [int(f.readline()) for x in range(n)]
ans = 1000000000
for i in range(n):
for j in range(i + 1, n):
for k in range(j + 1, n):
if (a[i] + a[j] + a[k]) % 4 == 0:
ans = min(ans, a[i] + a[j] + a[k])
print(ans)
Решение 2 (эффективное)
f = open(’8.txt’)
n = int(f.readline())
a = [int(i) for i in f]
k = 4
# Список из минимальных чисел с определенным остатком от деления
t = [10 ** 10] * k
# Список из пар с минимальными суммами с определенным остатком
v = [[10 ** 10, 10 ** 10] for i in range(k)]
mn = 10 ** 10
for i in range(2, len(a)):
# Обрабатываем элемент на расстоянии 2 от текущего (через один слева)
ost1 = a[i - 2] % k
# Если он меньше прошлого с таким остатком - обновляем список
t[ost1] = min(t[ost1], a[i - 2])
# Обрабатываем средний элемент - на расстоянии 1 от текущего (слева)
ost2 = a[i - 1] % k
# Для каждого из минимальных первых элементов (для каждого остатка)
# создаём новые суммы с новым средним элементом и обновляем
# эти суммы в списке v, если они получились меньше прошлых
for j in range(k):
if t[j] < 10 ** 10:
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]) < 10 ** 10 * 2:
sm = a[i] + sum(v[ost3])
mn = min(mn, sm)
print(mn)