Имеется набор данных из 
целых положительных чисел. Рассматриваются все пары различных элементов последовательности. Необходимо определить количество пар элементов, которые находятся на расстоянии не менее 
(за расстояние между элементами принять разность их индексов), их сумма делится на 
и при этом хотя бы один элемент из пары больше 
.
В первой строке файла находится число 
, в следующих 
строках даны элементы последовательности, целые положительные числа.
В ответ укажите два числа через пробел: сначала искомое значение для файла A, затем для файла B.
Переборное решение:
f = open("22_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]) % 71 == 0 and j - i >= 8:
cnt += 1
print(cnt)
Эффективное решение:
f = open(’22_A.txt’)
s = 8 # Расстояние между элементами
k = 71 # Чему должна быть кратна сумма
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)