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