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

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

Входные данные

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

Пример организации исходных данных во входном файле:

7

1

3

4

93

8

5

95

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

Предупреждение: для обработки файла B не следует использовать переборный алгоритм, вычисляющий сумму для всех возможных вариантов, поскольку написанная по такому алгоритму программа будет выполняться слишком долго.

with open(’1.txt’) as f:
    n = int(f.readline())
    k = [float(’inf’)]*31
    l = [0]*31
    s = 0
    m = 0
    minLen = float(’inf’)
    for i in range(n):
        x = int(f.readline())
        s += x
        if s % 31 == 0:
            if s > m:
                m = s
                minLen = i + 1
        ost = s % 31
        if k[ost] != float(’inf’):
            if (s-k[ost]) > m or (s-k[ost] == m and i - l[ost] + 1 < minLen):
                m = s - k[ost]
                minLen = i-l[ost] + 1
        if k[ost] == float(’inf’):
            k[ost] = s
            l[ost] = i + 1
        if s < k[ost]:
            k[ost] = s
            l[ost] = i + 1
print(minLen)

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