Даны два файла (27А.txt, 27B.txt), содержащие последовательность целых чисел. Определите количество пар элементов этих последовательностей (элементы пары не обязаны стоять в последовательности рядом, пара это два различных элемента последовательности), сумма которых не превышает , при этом первый элемент пары должен быть больше второго.
Входные данные:
Даны два входных файла: файл A и файл B, каждый из которых содержит в первой строке количество чисел N (). Каждая из следующих N строк содержит натуральное число, не превышающее
.
В ответе укажите два числа без пробела: сначала количество пар для файла А, затем для файла В.
Неэффективный алгоритм:
file = open(’27B.txt’)
n = int(file.readline())
arr = [int(_) for _ in file]
ans = 0
for i in range(n):
for j in range(i + 1, n):
if arr[i] + arr[j] <= 2678 and arr[i] > arr[j]:
ans += 1
print(ans)
Эффективный алгоритм:
file = open(’27B.txt’)
n = int(file.readline())
arr = [int(_) for _ in file]
# Список с количествами чисел от 0 до 2678
arr_summ = [0] * 2679
ans = 0
for i in range(n):
# Если число X не больше 2678, его можно поставить в пару с
# любым числом от 0 до (2678 - X) включительно
if arr[i] <= 2678:
# Увеличиваем количество пар на сумму количеств подходящих чисел
ans += sum(arr_summ[arr[i] + 1:2679 - arr[i]])
arr_summ[arr[i]] += 1
print(ans)