1989 - Teatru

De la Universitas MediaWiki

Alina este mare iubitoare de teatru. Directorul teatrului i-a oferit șansa să joace în mai multe spectacole, ca figurant, deocamdată. Costumiera de scenă a decis să-i dea C costume diferite dintre cele care sunt destinate acestei stagiuni. Alina va duce costumele acasă și le va ajusta ca să-i vină bine. Stagiunea durează Z zile consecutive și în fiecare zi se joacă câte o piesă. Aceeași piesă se va juca, desigur în una sau mai mai multe zile ale stagiunii. Fiecărei piese i se asociază un unic costum de figurant, deci pentru fiecare piesă în care joacă, Alina trebuie să îmbrace un singur costum, acela asociat piesei respective. Costumele de figuranți sunt identificate prin literele mari ale alfabetului englez: A, B, C, …, X, Y, Z. Alina are voie să-și aleagă cele C costume diferite.

Cerința

Cunoscând costumul asociat fiecărei zile a stagiunii, ajutați-o pe Alina să-și aleagă cele C costume diferite, în așa fel încât să poată juca într-un număr cât mai mare de piese consecutive.

Date de intrare

Fișierul de intrare teatru.in conține pe prima linie două numere naturale Z și C despărțite printr-un spațiu. Z este numărul de zile din stagiune iar C este numărul de costume diferite pe care Alina le poate primi. Pe linia a doua, se găsesc Z caractere, litere mari ale alfabetului englez. Caracterul al i-lea identifică costumul de figurant care trebuie îmbrăcat în spectacolul din ziua i.

Date de ieșire

Fișierul de ieșire teatru.out va conține pe prima linie un număr natural N, reprezentând numărul maxim de spectacole consecutive în care Alina poate juca. Pe linia a doua se scriu N caractere, fără spații între ele, corespunzătoare costumelor care vor fi îmbrăcate în cele N piese de teatru alese, în ordinea spectacolelor în care va juca. Dacă există mai multe soluții de lungime N atunci se afișează cea căreia îi corespunde o zi de început mai aproape de începutul stagiunii.

Restricții și precizări

  • 1 ≤ Z ≤ 55000
  • 1 ≤ C ≤ 26
  • C ≤ Z
  • Pentru 20% dintre teste Z ≤ 350

Exemplul 1

teatru.in

10 3
XKUFKFEGXG

teatru.out

5
KUFKF

Exemplul 2

teatru.in

15 4
IAJRAMRCZJJCDNS

teatru.out

6
AJRAMR

Încărcare soluție

Lipește codul aici

def read_input():
    global n, k, s
    n, k = map(int, input().split())
    s = input().strip()

def print_output():
    print(Lmax)
    print(''.join(s[imax:jmax+1]))

def calculate():
    global imax, jmax, Lmax
    i = j = nr = 0
    f = {chr(c): 0 for c in range(65, 91)}
    f[s[0]] += 1
    nr += 1
    while i < n and j < n:
        while nr <= k and j < n:
            j += 1
            if j >= n:
                break
            if f[s[j]] == 0:
                f[s[j]] += 1
                nr += 1
            else:
                f[s[j]] += 1
        if j - i > Lmax:
            Lmax = j - i
            imax = i
            jmax = j - 1
        while nr > k:
            f[s[i]] -= 1
            if f[s[i]] == 0:
                nr -= 1
            i += 1

with open('teatru.in', 'r') as fin:
    n, k = map(int, fin.readline().split())
    s = fin.readline().strip()

imax = jmax = Lmax = 0
calculate()

with open('teatru.out', 'w') as fout:
    fout.write(f"{Lmax}\n")
    fout.write(''.join(s[imax:jmax+1]) + '\n')

if __name__ == '__main__':
  calculate()