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