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