Editing 0961 - Criptare (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>

# Funcție care verifică dacă datele de intrare sunt valide
def validare_date(n: int, a: list[list[int]]) -> bool:

    if n < 1 or n > 1000:
        return False
    for i in range(n):
        if len(a[i]) != n:
            return False
        for j in range(n):
            if a[i][j] not in [0, 1]:
                return False
    return True

# Funcție care calculează numărul minim de calculatoare care trebuie criptate astfel încât hackerul Gigel să nu se mai
# se poată distra cu rețeaua.
def criptare(n: int, a: list[list[int]]) -> tuple[int, list[tuple[int, int]]]:
    L, C = [], []
    for i in range(n):
        linie_criptata = any(a[i][j] for j in range(n))  # verifică dacă există cel puțin un element 1 în linia i
        coloana_criptata = any(a[j][i] for j in range(n))  # verifică dacă există cel puțin un element 1 în coloana i
        if not linie_criptata:
            L.append(i + 1)  # adaugă linia i+1 la lista L dacă aceasta nu are niciun element 1
        if not coloana_criptata:
            C.append(i + 1)  # adaugă coloana i+1 la lista C dacă aceasta nu are niciun element 1
    numar_calculatoare_criptate = max(len(L), len(C))  # determină numărul minim de calculatoare care trebuie criptate
    puncte_criptare = [(L[i], C[i]) for i in range(min(len(L), len(C)))]  # perechile (linie, coloană) pentru primele L sau C elemente
    if len(L) > len(C):
        for i in range(len(C), len(L)):
            puncte_criptare.append((L[i], 1))  # adaugă perechi (linie, 1) pentru fiecare element L care nu are corespondent în C
    elif len(C) > len(L):
        for i in range(len(L), len(C)):
            puncte_criptare.append((1, C[i]))  # adaugă perechi (1, coloană) pentru fiecare element C care nu are corespondent în L
    return numar_calculatoare_criptate, puncte_criptare


#Programul principal
if __name__ == '__main__':
    with open('criptare.in', 'r') as fin:
        n = int(fin.readline().strip())
        a = [list(map(int, fin.readline().split())) for _ in range(n)]
    if validare_date(n, a):
        numar_calculatoare_criptate, puncte_criptare = criptare(n, a)
        with open('criptare.out', 'w') as fout:
            fout.write("Datele sunt introduse corect.\n")
            fout.write(str(numar_calculatoare_criptate) + '\n')
            for Sol in puncte_criptare:
                fout.write(str(Sol[0]) + ' ' + str(Sol[1]) + '\n')
    else:
        with open('criptare.out', 'w') as fout:
           fout.write("Datele nu corespund restricțiilor impuse.")

</syntaxhighlight>