Editing 2525 - Cioc (section)

== Rezolvare ==
<syntaxhighlight lang="python" line="1">
# 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")

</syntaxhighlight>