Дана последовательность натуральных чисел. Рассматриваются все её непрерывные подпоследовательности, в которых количество простых чисел кратно 10. Найдите наибольшую сумму такой подпоследовательности.
Входные данные: Даны два входных файла: файл A и файл B, каждый из которых содержит в первой строке количество чисел N . Каждая из следующих N строк содержит натуральное число, не превышающее
.
В ответе укажите два числа через пробел: сначала значение искомой суммы для файла А, затем для файла B.
def simple(x):
if x < 2:
return False
for i in range(2, int(x ** 0.5) + 1):
if x % i == 0:
return False
return True
f = open(’27B_09_5.txt’)
n = int(f.readline())
a = [int(i) for i in f]
# Чему кратно количество чисел
d = 10
r = 0 # Количество встреченных простых чисел
# Для сумм с нулевым количеством нужных чисел изначально самая маленькая - начальная
tails = [0] + [10 ** 10] * (d - 1)
mx = 0
s = 0
for i in range(n):
x = a[i]
# Увеличиваем общую сумму
s += x
# Если нашли нужное число - увеличиваем на 1 их количество
# перезаписываем хвост, выбирая наименьший для этого
# количества оканчивающихся k чисел
if simple(x):
r += 1
if s < tails[r % d]:
tails[r % d] = s
# Находим максимальную сумму, отсекая хвост с количеством оканчивающихся k чисел
# равным остатку от деления на необходимое количество d
if s - tails[r % d] > mx:
mx = s - tails[r % d]
print(mx)
Ответ: 464173 500516847