Имеется набор данных из 
целых чисел. Рассматриваются все пары различных элементов последовательности. Необходимо определить количество таких пар, сумма которых кратна 
и ровно одно число из пары больше 
.
В первой строке входных данных задаётся количество чисел 
. В каждой из последующих 
строк записано одно целое положительное число, не превышающее 
.
Пример входных данных:
5
9009
0
1
9
9018
Выходные данные для приведённого выше примера: 4
В ответе укажите через пробел два числа: сначала значение искомого количества для файла А, затем для файла B.
Переборное решение:
f = open(’14_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]) % 9 == 0:
# Критерии превышения 9000 у чисел отличаются,
# значит одно из них больше 9000, а другое нет
if (nums[i] > 9000) != (nums[j] > 9000):
cnt += 1
print(cnt)
Эффективное решение:
f = open("fileA.txt")
n = int(f.readline())
a = [int(x) for x in f.readlines()]
ost = [[0] * 2 for _ in range(9)]
# ost[i][0] - элемент <= 9000
# ost[i][1] - элемент > 9000
ans = 0
for i in range(n):
# Если элемент больше 9000, к нему в пару можно ставить только числа меньше 9000
if a[i] > 9000:
ans += ost[(9 - a[i] % 9) % 9][0]
ost[a[i] % 9][1] += 1
# Иначе к нему в пару можно ставить только числа больше 9000
else:
ans += ost[(9 - a[i] % 9) % 9][1]
ost[a[i] % 9][0] += 1
print(ans)
f.close()