На вход подается число 
, а затем последовательность из 
натуральных чисел. Напишите программу, которая находит количество пар элементов, произведение которых кратно 
и хотя бы один элемент из пары меньше 
, при условии, что элементы стоят на расстоянии не меньше 
, то есть 
, где 
— номера элементов последовательности.
В первой строке файла находится число 
, в следующих 
строках даны элементы последовательности, целые положительные числа.
В ответ укажите два числа через пробел: сначала искомое значение для файла A, затем для файла B.
Переборное решение:
f = open("24_A.txt")
n = int(f.readline())
a = [int(i) for i in f]
ans = 0
for i in range(n):
for j in range(i + 1, n):
if a[i] * a[j] % 4 == 0 and (a[i] < 20 or a[j] < 20) and j - i >= 3:
ans += 1
print(ans)
Эффективное решение:
f = open(’24_B.txt’)
s = 3 # Расстояние между элементами
p = 4 # Чему должно быть кратно произведение
d = [4, 2, 1]
n = int(f.readline())
a = [int(i) for i in f]
# Список с количествами чисел, удовлетворяющих определенным условием
# Индексы числа nums[x][y] обозначают следующее:
# x - меньше ли число чем 20 (1 - да, 0 - нет)
# y - максимальный делитель p, которому кратно число
nums = [[0 for _ in range(p + 1)] for _ in range(2)]
ans = 0
for i in range(s, n):
# Получаем индекс для первого критерия
ind = int(a[i - s] < 20)
# Ищем максимальный делитель, которому кратно число
for j in d:
if a[i - s] % j == 0:
# Обновляем максимальное число с такими характеристиками
nums[ind][j] += 1
break
# Перебираем все возможные делители и ищем те, при которых произведение будет кратно 4
for j in d:
if a[i] % j == 0:
# Обновляем ответ, если новая сумма больше прошлой
ans += nums[1][p // j]
# Если a[i] меньше 20, то его можно поставить в пару с числами больше 20
if a[i] < 20:
ans += nums[0][p // j]
print(ans)