27.08 Макс/мин, кол-во пар, сумма/разность/произведение кратно/не кратно на расстоянии

Решение 1 (неэффективное)

f = open(’15_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]) % 61 == 0 and (a[i] * a[j]) % 2 != 0 and j - i >= 3:
            cnt += 1

print(cnt)

Решение 2 (эффективное)

f = open(’15_B.txt’)

n = int(f.readline())
a = [int(i) for i in f]
cnt = 0

k = 61  # Чему должна быть кратна сумма
s = 3  # Расстояние между элементами

# Список количеств чисел с определенными остатками от деления на k
# Например, под индексом 3 хранится кол-во чисел с остатком 3
nums = [0] * k

for i in range(s, n):
    # Чтобы произведение было нечетным, оба числа должны быть нечетными
    # поэтому работаем только с нечетными числами

    if a[i - s] % 2 != 0:
        # Считаем остаток от деления первого числа пары на k
        ost1 = a[i - s] % k
        # Увеличиваем кол-во чисел с найденным остатком
        nums[ost1] += 1

    if a[i] % 2 != 0:
        # Считаем остаток, который должен быть у числа,
        # которое можно поставить в пару с текущим
        ost2 = (k - (a[i] % k)) % k
        # Увеличиваем ответ на количество образованных пар
        cnt += nums[ost2]

print(cnt)