ЕГЭ №24

Найдём позиции всех букв C. Добавим фиктивные позиции -1 и len(s), чтобы удобно обрабатывать начало и конец строки.

Будем брать каждые 30 подряд идущих букв C. Для такого набора левая граница фрагмента может находиться после предыдущей C, а правая граница — перед следующей C.

Так как фрагмент должен начинаться и заканчиваться на A, нужно перемножить количество букв A в левой допустимой зоне и количество букв A в правой допустимой зоне.

s = open('24_A_30_C_01.txt').read().strip()

c = [-1]
for i, ch in enumerate(s):
    if ch == 'C':
        c.append(i)
c.append(len(s))

ans = 0
k = 30

for i in range(1, len(c) - k):
    left_zone_l = c[i - 1] + 1
    left_zone_r = c[i]

    right_zone_l = c[i + k - 1]
    right_zone_r = c[i + k] - 1

    left_a = s[left_zone_l:left_zone_r + 1].count('A')
    right_a = s[right_zone_l:right_zone_r + 1].count('A')

    ans += left_a * right_a

print(ans)

Ответ: 3047.

Будем идти по строке справа налево? Нет, удобнее идти слева направо и поддерживать окно — текущий фрагмент строки.

Переменная left хранит левую границу фрагмента, а count_a — количество букв A внутри него.

s = open('24_50_A_01.txt').read().strip()

left = 0
count_a = 0
best = 0

for right, ch in enumerate(s):
    if ch == 'A':
        count_a += 1

    while count_a > 50:
        if s[left] == 'A':
            count_a -= 1
        left += 1

    best = max(best, right - left + 1)

print(best)

Ответ: 4225.

Будем просматривать строку слева направо. Переменная left хранит начало текущего допустимого фрагмента, а a_run — сколько букв A подряд встретилось к текущему моменту.

Если a_run стало равно 3, значит текущий фрагмент уже содержит запрещённую цепочку AAA. Тогда переносим левую границу так, чтобы в новом фрагменте остались только две последние буквы A.

s = open('24_no_A_01.txt').read().strip()

best = 0
left = 0
a_run = 0

for right, ch in enumerate(s):
    if ch == 'A':
        a_run += 1
    else:
        a_run = 0

    if a_run == 3:
        left = right - 1
        a_run = 2

    best = max(best, right - left + 1)

print(best)

s = open('24_A1000.txt').read().strip()

ans = 0
current = 0

for ch in s:
    if ch == 'A':
        current += 1
    else:
        ans += current // 10
        current = 0

ans += current // 10

print(ans)