Дана последовательность из N натуральных чисел. Рассматриваются все её непрерывные подпоследовательности, такие что сумма элементов каждой из них кратна k = 31. Найдите среди них подпоследовательность с максимальной суммой, определите её длину. Если таких подпоследовательностей найдено несколько, в ответе укажите количество элементов самой короткой из них.
Входные данные
Даны два входных файла (файл A и файл B), каждый из которых содержит в первой строке количество чисел N . Каждая из следующих 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)