Дана последовательность целых положительных чисел. Необходимо определить количество пар элементов этой последовательности, сумма которых делится на
и при этом хотя бы один элемент из пары больше
.
Входные данные.
Даны два входных файла («file_A»и «file_B»), каждый из которых содержит в первой строке количество чисел (
). В каждой из последующих
строк записано одно натуральное число, не превышающее 1000.
В ответе укажите два числа: сначала значение искомого количества для файла А, затем для файла B.
Решение 1 (неэффективное)
f = open("27A.txt")
n = int(f.readline())
a = [int(f.readline()) for x in range(n)]
ans = 0
for i in range(n):
for j in range(i + 1, n):
if (a[i] + a[j]) % 42 == 0:
if (a[i] > 512) or (a[j] > 512):
ans += 1
print(ans)
Решение 2 (эффективное)
file = open(’file_B.txt’)
n = int(file.readline())
ans = 0
# Список чисел с определенными остатками от деления, которые больше 512
k_over_512 = [0] * 42
# Список чисел с определенными остатками от деления, которые меньше 512
k_less_512 = [0] * 42
for i in range(n):
x = int(file.readline())
ost = x % 42
dop = (42 - ost) % 42
# Если x больше 512, к нему в пару можно ставить числа как больше,
# так и меньше 512
if x > 512:
ans += k_over_512[dop] + k_less_512[dop]
k_over_512[ost] += 1
# Если x меньше 512, к нему в пару можно ставить только числа больше 512
else:
ans += k_over_512[dop]
k_less_512[ost] += 1
print(ans)