Дана последовательность из N натуральных чисел. Рассматриваются все её непрерывные подпоследовательности, такие что сумма элементов каждой из них кратна k = 317. Найдите среди них подпоследовательность с максимальной суммой, определите её длину. Если таких подпоследовательностей найдено несколько, в ответе укажите количество элементов самой короткой из них.
Входные данные: Даны два входных файла: файл A и файл B, каждый из которых содержит в первой строке количество чисел N (
). Каждая из следующих N строк содержит натуральное число, не превышающее 10 000.
пример входного файла:
7
104
127
86
119
166
151
195
В этом наборе можно выбрать последовательности 104+127+86 (сумма 317) и 166+151 (сумма 317). Самая короткая из них имеет длину 2. Ответ: 2.
В ответе укажите два числа через пробел: сначала значение искомой суммы для файла А, затем для файла B.
with open(’5_B.txt’) as f:
n = int(f.readline())
k = [float(’inf’)]*317
l = [0]*317
s = 0
m = 0
minLen = float(’inf’)
for i in range(n):
x = int(f.readline())
s += x
if s % 317 == 0:
if s > m:
m = s
minLen = i + 1
ost = s % 317
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)