Задача к ЕГЭ по информатике на тему «Цепочки, выбор последовательности, префиксные суммы» №9

Дана последовательность из N натуральных чисел. Рассматриваются все её непрерывные подпоследовательности, такие что сумма элементов каждой из них кратна k = 97 и при этом количество чисел кратных 13 равно 17. Найдите среди них подпоследовательность с максимальной суммой. В ответ укажите значение суммы.

Входные данные: Даны два входных файла: файл A и файл B, каждый из которых содержит в первой строке количество чисел N (2 ≤ N ≤ 108  ). Каждая из следующих N строк содержит натуральное число, не превышающее 10 000.

В ответе укажите два числа через пробел: сначала значение искомой суммы для файла А, затем для файла B.

Решение для А пункта:
f = open(’5_27A.txt’)
n = int(f.readline())
a = [int(i) for i in f]
mx = 0
for i in range(len(a)):
    s = 0
    k = 0
    for j in range(i,len(a)):
        s += a[j]
        if a[j] % 13 == 0:
            k += 1
        if s % 97 == 0 and k == 17:
            mx = max(mx,s)
print(mx)
Решение для Б пункта:
f = open(’5_27B.txt’)
n = int(f.readline())
D = 97
M = 13
d = {(x,y):10**20 for x in range(D) for y in range(-17,76828+1)}
#Предварительно, нужно посчитать количество чисел в файле,
#кратных 13. Перебор начинается от -17 во избежании ошибки,
#поскольку на первых итерациях значение выражения k - 17 будет отрицательным.
mx = 0
s = 0
k = 0
for i in range(n):
    x = int(f.readline())
    s += x
    if x % M == 0:
        k += 1
    if s % D == 0 and k == 17:
        mx = max(mx,s)
    s1 = s - d[(s % D,k-17)]#получаем сумму, которая делится нацело на 97
    # и содержит ровно 17 чисел кратных 13
    mx = max(mx,s1)
    d[(s%D,k)] = min(d[(s%D,k)],s)#обновление значений в словаре
print(mx)

Ответ: 125033 2500854
Оцените статью
Я решу все!