На вход программы поступает последовательность из 
натуральных чисел. Необходимо найти количество пар различных элементов последовательности, находящихся на расстоянии не менее 
, чтобы их сумма делилась на 
и хотя бы один элемент пары должен быть меньше 
.
Даны два входных файла (файл A и файл B), каждый из которых содержит в первой строке количество чисел 
(
). Каждая из следующих 
строк содержит одно натуральное число, не превышающее 
.
В ответе укажите два числа: сначала найденное количество искомых пар для файла А, затем для файла B.
Переборное решение:
f = open("19_A.txt")
n = int(f.readline())
a = []
for i in range(n):
a.append(int(f.readline()))
cnt = 0
for i in range(n):
for j in range(i + 1, n):
if (a[i] < 35 or a[j] < 35) and (a[i] + a[j]) % 13 == 0 and j - i >= 3:
cnt += 1
print(cnt)
Эффективное решение:
f = open(’19_B.txt’)
s = 3 # Расстояние между элементами
k = 13 # Чему должна быть кратна сумма
n = int(f.readline())
a = [int(i) for i in f]
# Список с количествами чисел, удовлетворяющих определенным условием
# Индексы числа nums[x][y] обозначают следующее:
# x - меньше ли число чем 35 (1 - да, 0 - нет)
# y - остаток от деления числа на k
nums = [[0] * k for _ in range(2)]
cnt = 0
for i in range(s, n):
# Находим остаток первого числа пары
ost1 = a[i - s] % k
# Увеличиваем количество чисел с такими характеристиками
nums[int(a[i - s] < 35)][ost1] += 1
# Вычисляем остаток от деления на k числа-пары для x
ost2 = (k - (a[i] % k)) % k
# Увеличиваем ответ на количество пар с x, сумма которых будет кратна k,
# где первое число точно меньше 35
cnt += nums[1][ost2]
# Если a[i] меньше 35, то его можно поставить также в пару с числами больше 35
if a[i] < 35:
cnt += nums[0][ost2]
print(cnt)