1133 - Charlie

Enunt

Charlie a decis să se joace cu literele dintr-un șir de caractere, șir ce conține doar literele mici ale alfabetului englez 'a'...'z'. Jocul constă în a elimina litere din șir după următoarea regulă: fie L1, L2, L3 trei litere aflate pe poziții consecutive în șir, atunci litera L2 poate fi eliminată dacă și numai dacă este strict mai mică lexicografic decât literele L1 și L3.

Pentru a face jocul mai interesant, Charlie atașează eliminării literei L2 un cost egal cu valoarea maximă dintre ō(L1) și ō(L3), unde prin ō(litera) înțelegem numărul de ordine al literei respective în alfabet (ō('a') = 1, ō('b') = 2, …, ō('z') = 26). Charlie aplică în mod repetat procedeul de eliminare și calculează suma costurilor eliminărilor efectuate.

Cerinţe

Fiind dat un șir de caractere să se determine:

a) Lungimea maximă a unei secvențe de litere alternante, adică o secvență pentru care literele aflate pe poziții consecutive sunt de forma: Li > Li+1 < Li+2 > Li+3 < Li+4 > … < Lj.

b) Suma maximă pe care o poate obține Charlie aplicând în mod repetat procedeul de eliminare a literelor, precum și șirul obținut în final.

Date de intrare

Fișierul de intrare charlie.in conține pe prima linie numărul n, iar pe a doua linie n numere naturale separate prin spații.

Date de ieșire

Dacă valoarea lui p este 1, se va rezolva numai punctul a) din cerință.

În acest caz, în fişierul de ieşire charlieIN.txt se va scrie un singur număr natural L ce reprezintă lungimea maximă a unei secvențe de litere alternante.

Dacă valoarea lui p este 2, se va rezolva numai punctul b) din cerință.

În acest caz, fişierul de ieşire charlieOUT.txt va conține două linii. Pe prima linie se va afla șirul rezultat în urma eliminărilor repetate de litere respectând regula enunțată, iar pe cea de-a doua linie suma maximă obținută. În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Datele nu corespund restrictiilor impuse".

Restricții și precizări

• 3 ≤ numărul de litere ale șirului inițial ≤ 100000
• Pentru rezolvarea corectă a primei cerinţe se acordă 25 de puncte, iar pentru cerința a doua se acordă 75 de puncte. Pentru 30% dintre teste numărul de litere ale șirului ≤ 1000

Exemplul 1:

charlieIN.txt

1
cadgfacbda

charlieOUT.txt

5

Explicație

p = 1

Secvențele alternante corect formate sunt: cad, facbd. Lungimea maximă este 5.

Exemplul 1:

charlieIN.txt

1
ca

charlieOUT.txt

Datele nu corespund restrictiilor impuse

Rezolvare

ordine = 96

def verify_restrictions(length):
    if length < 3 or length > 100000:
        return "Datele nu corespund restrictiilor impuse"
    return None

def read():
    with open("charlieIN.txt", "r") as cin:
        lines = cin.readlines()
        task = int(lines[0])
        v = [ord(c) for c in lines[1].strip()]  # Convert characters to their ASCII values
        n = len(v)

        # Verificare restricții
        restriction_error = verify_restrictions(n)
        if restriction_error:
            with open("charlieOUT.txt", "w") as cout:
                cout.write(restriction_error)
            return None, None, None

        return task, v, n

def solve1(v, n):
    lmax = l = t = 0
    start = False
    i = 0
    t = 1

    while i < n - 1:
        if not start:
            if v[i] * t > v[i + 1] * t:
                start = True
                l = 2
                t *= -1
            i += 1
        else:
            if v[i] * t > v[i + 1] * t:
                l += 1
                t *= -1
                i += 1
            else:
                if t == -1:
                    l -= 1
                lmax = max(lmax, l)
                l = 0
                start = False
                t = 1

    if l and t == -1:
        l -= 1

    lmax = max(lmax, l)

    return lmax

def parcurgere(v, n):
    i = k = r = 0
    modificari = False
    new_v = [v[0]]
    cost = 0

    i = 1
    while i < n - 1:
        while i < n - 1 and v[i] < min(new_v[k], v[i + 1]):
            cost += max(new_v[k], v[i + 1]) - ordine
            modificari = True
            i += 1
            r += 1
        new_v.append(v[i])
        k += 1
        i += 1

    new_v.append(v[n - 1])
    n = len(new_v)

    return modificari, new_v, n, cost

def solve2(v, n):
    ok = True
    total_cost = 0

    while ok:
        ok, v, n, cost = parcurgere(v, n)
        total_cost += cost

    return ''.join([chr(c) for c in v]), total_cost

def main():
    task, v, n = read()
    if task is not None:
        if task % 2:
            result = solve1(v, n)
        else:
            result, cost = solve2(v, n)
            result = f"{result}\n{cost}"
        
        with open("charlieOUT.txt", "w") as cout:
            cout.write(str(result))

if __name__ == "__main__":
    main()