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

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

Даны два входных файла (файл А и файл В), каждый из которых содержит в первой строке одно натуральное число N (1 ≤ N ≤ 1000000)  — количество чисел. Каждая из следующих N  строк содержит целое число. В ответе запишите максимальную сумму кратную трем сначала для файла A, затем для файла B через пробел.

Решение 1 (неэффективное)

f = open(’27.txt’)
n = int(f.readline())
a = [int(f.readline()) for i in range(n)]
maxim = 0
for i in range(n):
    s = 0
    for j in range(i, n):
        s += a[j]
        if s % 3 == 0:
            maxim = max(maxim, s)
print(maxim)

Переборное решение будет обрабатывать более 90000  чисел в файле слишком долго, поэтому напишем эффективное решение.

 

Решение 2 (эффективное)

f = open(’2_27B__vxok.txt’)
n = int(f.readline())
mx = 0 # максимальная сумма подпоследовательности кратной 3
s = 0 # сумма всех чисел файла
D = 3 # наш делитель
tails = [10**30]*D # префиксные суммы под определенным остатком при делении на 3
for i in range(n):
    x = int(f.readline()) # считываем текущее число
    s += x
    if s % D == 0 and s > mx: # если сумма кратна 3 и больше максимума
        mx = s # перезаписываем максимальную сумму
    s1 = s - tails[s % D] # если сумма s не кратна 3,
    # то убавляем префиксную сумму с остатком при делении на 3 равным остатку при делении на 3 у суммы s
    # тогда мы получим новую сумму s1, которая кратна 3
    # если сумма s1 больше максимума
    mx = max(mx,s1) # перезаписываем максимум
    # если сумма s меньше той, что записана в списке tails с таким же остатком при делении на 3 как и у s
    tails[s % D] = min(s,tails[s % D]) # перезаписываем сумму
print(mx)

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