Имеется набор данных из 
целых положительных чисел. Рассматриваются все пары различных элементов последовательности. Необходимо определить количество пар элементов,которые находятся на расстоянии не меньше 
, их сумма делится на 
и при этом хотя бы один элемент из пары больше 
.
В первой строке входных данных задаётся количество чисел 
. В каждой из последующих 
строк записано одно целое положительное число, не превышающее 
.
В ответе укажите два числа через пробел: сначала значение искомой суммы для файла 
, затем для файла 
.
Переборное решение:
f = open("9_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] > 21 or a[j] > 21) and (a[i] + a[j]) % 74 == 0 and j - i >= 5:
cnt += 1
print(cnt)
Эффективное решение:
f = open(’9_B.txt’)
s = 5 # Расстояние между элементами
k = 74 # Чему должна быть кратна сумма
n = int(f.readline())
a = [int(i) for i in f]
# Список с количествами чисел, удовлетворяющих определенным условием
# Индексы числа nums[x][y] обозначают следующее:
# x - больше ли число чем 21 (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] > 21)][ost1] += 1
# Вычисляем остаток от деления на k числа-пары для x
ost2 = (k - (a[i] % k)) % k
# Увеличиваем ответ на количество пар с x, сумма которых будет кратна k,
# где первое число точно больше 21
cnt += nums[1][ost2]
# Если a[i] больше 21, то его можно поставить также в пару с числами меньше 21
if a[i] > 21:
cnt += nums[0][ost2]
print(cnt)