На вход программы поступает последовательность из N целых положительных чисел. Рассматриваются все пары различных элементов последовательности, в которых между любыми двумя элементами расстояние не более 28 (расстояние – разность порядковых номеров). Необходимо определить максимальную сумму элементов пары, кратную 777.
Входные данные: Даны два входных файла, каждый из которых содержит в первой строке количество пар N . Каждая из следующих N строк содержит одно натуральное число.
Переборное решение:
f = open(’A.txt’)
n = int(f.readline())
a = [int(i) for i in f]
ans = -10 ** 10
k = 777 # Чему должна быть кратна сумма
s = 28 # Расстояние между элементами
for i in range(n):
for j in range(i + 1, n):
if (a[j] + a[i]) % k == 0 and j - i <= s:
ans = max(ans, a[i] + a[j])
print(ans)
Эффективное решение:
f = open(’5_27B_pairs.txt’)
n = int(f.readline())
a = [int(i) for i in f]
d = 28
k = 777
# Список из элементов на нужном расстоянии
lst = []
mx = -1
for i in range(len(a)):
# Считываем текущий элемент
a1 = a[i]
# Проходимся по d элементов до текущего
for a2 in lst:
# Если сумма удовлетворяет условию - обновляем максимум
if (a1 + a2) % k == 0:
mx = max(mx, a1 + a2)
# Добавляем текущий элемент в массив
lst.append(a1)
# Убираем элемент, который будет слишком далеко для нового элемента
if len(lst) > d:
lst.pop(0)
print(mx)