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