Обработка массива: задания №17 и №25

with open('17-data.csv') as f:
    nums = [int(line) for line in f]

# Если перебирать все пары чисел, это O(n^2): на файле с миллионом строк —
# не уложиться во время экзамена. Сумма двух чисел кратна 5 только тогда,
# когда их остатки от деления на 5 в сумме дают 0 или 5 — то есть подходят
# только три комбинации остатков: (0,0), (1,4), (2,3). Достаточно один раз
# пройти по файлу и для каждого остатка запомнить, сколько чисел его дают
# и два самых больших числа среди них (второе — на случай пары 0+0).
count_by_rem = [0] * 5
top1 = [None] * 5
top2 = [None] * 5

for x in nums:
    r = x % 5
    count_by_rem[r] += 1
    if top1[r] is None or x > top1[r]:
        top1[r], top2[r] = x, top1[r]
    elif top2[r] is None or x > top2[r]:
        top2[r] = x

count = 0
max_sum = None

c0 = count_by_rem[0]
count += c0 * (c0 - 1) // 2
if c0 >= 2:
    max_sum = top1[0] + top2[0]

for r in (1, 2):
    other = 5 - r
    count += count_by_rem[r] * count_by_rem[other]
    if count_by_rem[r] and count_by_rem[other]:
        s = top1[r] + top1[other]
        if max_sum is None or s > max_sum:
            max_sum = s

print(count)
print(max_sum)