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