Editing 2110 - Vot (section)

=== Explicație ===
Se observă că <code>1</code> îl votează pe <code>10</code>, <code>3</code> se votează pe el însuşi, votul lui <code>7</code> merge spre <code>1</code>, al lui <code>9</code> – spre <code>7</code> iar <code>10</code> îl votează pe <code>9</code>, deci toti elevii din acest grup primesc voturi din partea unor elevi ai grupului şi acesta este maximal cu proprietatea cerută.<syntaxhighlight lang="python3">
import numpy as np

def main():
    # Deschidem fisierul de intrare "vot.in" pentru citire si fisierul de iesire "vot.out" pentru scriere.
    fin = open("vot.in", "r")
    fout = open("vot.out", "w")

    # Initializam array-uri pentru stocarea datelor
    numar_a = np.zeros(1001, dtype=int)
    numar_b = np.zeros(1001, dtype=int)
    numar_c = np.zeros(1001, dtype=int)

    # Citim numarul 'n' din prima linie a fisierului de intrare.
    numar_n = int(fin.readline())

    # Citim o linie de numere intregi separate prin spatiu si le stocheaza in array-ul 'a'.
    line = fin.readline().split()
    for i in range(1, numar_n + 1):
        # Converteste valorile citite din fisier in intregi si le stocheaza in 'a'.
        numar_a[i] = int(line[i - 1])
        # Incrementam contor pentru fiecare valoare din 'a' in array-ul 'b'.
        numar_b[numar_a[i]] += 1

    # Initializam pointeri si array suplimentar pentru a retine anumite valori.
    numar_p, numar_u = 1, 0

    # Identificam valorile care nu apar in 'a'.
    for i in range(1, numar_n + 1):
        if numar_b[i] == 0:
            numar_u += 1
            numar_c[numar_u] = i

    # Eliminam anumitor elemente din 'a' bazat pe contorul din 'b'.
    while numar_p <= numar_u:
        j = numar_c[numar_p]
        if numar_a[j] > 0 and numar_b[j] == 0:
            numar_b[numar_a[j]] -= 1
            if numar_b[numar_a[j]] == 0:
                numar_u += 1
                numar_c[numar_u] = numar_a[j]
            numar_a[j] = 0
        numar_p += 1

    # Scriem rezultatul in fisierul de iesire
    fout.write(str(numar_n - numar_u) + "\n")
    for i in range(1, numar_n + 1):
        if numar_a[i] > 0:
            fout.write(str(i) + " ")

    # Inchidem fisierele de intrare si iesire.
    fin.close()
    fout.close()

if __name__ == "__main__":
    main()
</syntaxhighlight>