Имеется набор данных из N целых чисел. Рассматриваются все пары различных элементов последовательности. Необходимо определить количество пар элементов, сумма которых делится на 91 и при этом хотя бы один элемент из пары строго больше 33.
В первой строке входных данных задаётся количество чисел N (1 N
10000). В каждой из последующих N строк записано одно целое положительное число, не превышающее 10 000.
Пример входных данных:
5
90
1
7
92
33
Выходные данные для приведённого выше примера: 2
В ответе укажите два числа: сначала значение искомого количества для файла А, затем для файла B.
Переборное решение:
f = open(’11_A.txt’)
n = int(f.readline())
nums = [int(i) for i in f]
cnt = 0
for i in range(n):
for j in range(i + 1, n):
if (nums[i] + nums[j]) % 91 == 0:
if nums[i] > 33 or nums[j] > 33:
cnt += 1
print(cnt)
Эффективное решение:
n = int(input())
# Количества чисел с определенными остатками, которые больше 33
b_33 = [0]*91
# Количества чисел с определенными остатками, которые меньше 33
m_33 = [0]*91
ans = 0
for i in range(n):
x = int(input())
# Если x больше 33, к нему в пару можно ставить числа
# и больше и меньше 33
if x > 33:
ans += b_33[(91 - x%91) % 91] + m_33[(91 - x%91) % 91]
b_33[x%91] += 1
# Иначе же к нему в пару можно ставить только числа больше 33
else:
ans += b_33[(91 - x%91) % 91]
m_33[x%91] += 1
print(ans)