Editing 1451 - Iceberg (section)

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

def validare_matrice(matrice):
    n = len(matrice)
    # verificăm dacă matricea este pătratică și are dimensiunea între 1 și 31
    if not all(len(row) == n for row in matrice) or n < 1 or n > 31:
        return False
    # verificăm dacă marginile matricei conțin doar valori de 0
    if not all(matrice[0][j] == 0 and matrice[n-1][j] == 0 and matrice[i][0] == 0 and matrice[i][n-1] == 0
               for i in range(n) for j in range(n)):
        return False
    # verificăm dacă matricea conține doar valori de 0 și 1
    if not all(val in [0, 1] for row in matrice for val in row):
        return False
    # verificăm dacă icebergul este înconjurat de apă
    for i in range(1, n-1):
        for j in range(1, n-1):
            if matrice[i][j] == 1 and all(matrice[i+di][j+dj] == 0 for di, dj in [(0,1), (1,0), (0,-1), (-1,0)]):
                return False
    return True

def topire_iceberguri(matrice):
    n = len(matrice)
    S = sum(row.count(1) for row in matrice)  # calculăm inițial numărul de celule din iceberg
    v = [S]  # adăugăm numărul inițial de celule în lista de soluții
    while S > 0:  # repetăm până când icebergul a fost complet topit
        for i in range(1, n-1):
            for j in range(1, n-1):
                if matrice[i][j] == 1:  # dacă celula este o parte din iceberg
                    contor = 0
                    if matrice[i-1][j] == 0: contor += 1  # dacă celula de sus este apă
                    if matrice[i+1][j] == 0: contor += 1  # dacă celula de jos este apă
                    if matrice[i][j-1] == 0: contor += 1  # dacă celula din stânga este apă
                    if matrice[i][j+1] == 0: contor += 1  # dacă celula din dreapta este apă
                    if contor >= 2:  # dacă icebergul are mai mult de o celulă de apă învecinată
                        S -= 1  # scădem numărul de celule din iceberg
                        matrice[i][j] = 2  # marcam celula curenta ca fiind topita
        v.append(S)  # adăugăm numărul de celule din iceberg în lista de soluții
        for i in range(1, n-1):
            for j in range(1, n-1):
                if matrice[i][j] == 2:
                    matrice[i][j] = 0  # resetăm celulele topite la valoarea de apă
    return v

if __name__ == '__main__':
   # citim matricea de la tastatură
    n = int(input())
    matrice = []
    for i in range(n):
        linie = list(map(int, input().split()))
        matrice.append(linie)

    # validăm matricea
    if not validare_matrice(matrice):
       print("Datele nu corespund restricțiilor impuse.")
    else:
       solutie = topire_iceberguri(matrice)
       print("Datele sunt introduse corect.")
       print(len(solutie)-1)
    for t in solutie[:-1]:
        print(t)


</syntaxhighlight>