2525 - Cioc

Cioc

Cioc, un elev abia aterizat în clasa a IX-a, primește de la doamna profesor de informatică un șir de n numere naturale pe care trebuie să îl prelucreze. Astfel, băiatul trebuie să scrie după fiecare dintre cele k cele mai mici elemente dublul lor. Dacă cel mai mare dintre aceste numere se repetă și deja se depășesc cele k elemente prevăzute, doamna profesor îi dă libertatea băiatului de a modifica valoarea lui k astfel încât să cuprindă și aceste valori. De exemplu, dacă n = 7, v[] = {1, 4, 6, 2, 3, 4, 5} și k = 4, atunci, în urma prelucrării, șirul v devine {1, 2, 4, 8, 6, 2, 4, 3, 6, 4, 8, 5}, și deci kfinal=5.

Cerința

Cunoscându-se n, șirul v și k, să se afișeze:

• 1. Numărul kfinal;
• 2. Vectorul după prelucrare.

Date de intrare

Fișierul de intrare ciocin.txt conține pe prima linie numerele c (mereu 1 sau 2, reprezentând cerința ce trebuie rezolvată), n, k, iar următoarea linie va conține n numere întregi, reprezentând elementele vectorului inițial.

Date de ieșire

Fișierul de ieșire ciocout.txt va conține pe prima linie:

• Dacă c = 1, atunci numărul kfinal;
• Dacă c = 2, atunci vectorul după prelucrare.

Restricții și precizări

• 1 ≤ n, k ≤ 100.000;
• 1≤v[i]≤2^60;
• Pentru 40% din teste, c=1.

Exemplul 1

ciocin.txt

1 7 4

1 4 6 2 3 4 5

ciocout.txt

Datele introduse corespund restricțiilor impuse.

5

Explicație

k-ul inițial nu este suficient de cuprinzător (numărul 4 se repetă).

Exemplul 2

ciocin.txt

2 7 4

1 4 6 2 3 4 5

ciocout.txt

Datele introduse corespund restricțiilor impuse.

1 2 4 8 6 2 4 3 6 4 8 5

Explicație

Acesta este exemplul din enunț.

Exemplul 3

ciocin.txt

2 7 4

1 4 6 2 3 4 1000000001

ciocout.txt

Datele introduse nu corespund restricțiilor impuse.

Rezolvare

# 2525 - Cioc
def validare(n_validare, k_validare, sir_validare):           # functia de validare a datelor de intrare
    if not (1 <= n_validare <= 100000 and 1 <= k_validare <= 100000):
        raise ValueError

    for v in sir_validare:
        if not (1 <= v <= 260):
            raise ValueError

    fisier_iesire.write("Datele de intrare corespund restrictiilor impuse\n")


def prelucrare(n_prelucrare, k_prelucrare, sir_prelucrare):         # functia de rezolvare a problemei
    sir_sortat = sorted([(v, i) for i, v in enumerate(sir_prelucrare)])
    k_final_prelucrare = k_prelucrare
    while k_final_prelucrare < n_prelucrare and sir_sortat[k_final_prelucrare][0] == sir_sortat[k_prelucrare-1][0]:
        k_final_prelucrare += 1

    sir_prelucrat_prelucrare = []
    for v, i in sir_sortat[:k_final_prelucrare]:
        sir_prelucrat_prelucrare.append((i, v))
        sir_prelucrat_prelucrare.append((i, v*2))
    sir_prelucrat_prelucrare.sort()

    return k_final_prelucrare, [v for i, v in sir_prelucrat_prelucrare]


if __name__ == '__main__':
    fisier_intrare = open("ciocin.txt", "r")         # declararea fisierelor
    fisier_iesire = open("ciocout.txt", "w")       # fisierul out trebuie declarat cu optiunea "w" (write)

    try:
        c, n, k = map(int, fisier_intrare.readline().split())
        sir = list(map(int, fisier_intrare.readline().split()))

        validare(n, k, sir)                 # apelul functiei de validare

        k_final, sir_prelucrat = prelucrare(n, k, sir)  # apelul functiei de rezolvare

        if c == 1:
            fisier_iesire.write(str(k_final) + "\n")
        else:
            fisier_iesire.write(" ".join(map(str, sir_prelucrat)) + "\n")

    except ValueError:
        fisier_iesire.write("Datele de intrare nu corespund restrictiilor impuse")