0882 - Lac

Cerința

Se dă harta unui lac de formă dreptunghiulară, împărțit în n*m zone dispuse sub forma unei matrice cu n linii și m coloane. Zonele pot fi acoperite cu apă, sau pot fi zone de uscat. Zonele de uscat care sunt învecinate pe linie sau pe coloană formează insule sau peninsule. Peninsule conțin cel puțin o zonă de uscat pe marginea lacului (matricei), în timp ce insulele sunt situate în întregime în interiorul lacului.

Cunoscând harta lacului, determinați numărul de insule și numărul de peninsule.

Date de intrare

Fișierul de intrare lacin.txt conține pe prima linie numerele n m. Următoarele n linii conțin câte m valori 0 sau 1, separate prin câte un spațiu. Valoarea 0 reprezintă o zonă acoperită cu apă, iar valoarea 1 reprezintă o zonă acoperită cu uscat.

Date de ieșire

Fișierul de ieșire lacout.txt va conține pe prima două numerele nrI nrP, separate prin exact un spațiu, reprezentând numărul de insule, respectiv numărul de peninsule existente pe lac.

Restricții și precizări

• 1 ≤ n , m ≤ 1000

Exemplul 1:

lac.in

6 9
1 1 0 0 1 1 1 0 0 
0 0 1 0 1 0 1 0 0 
1 1 0 0 0 0 0 0 0 
1 0 0 0 1 1 1 0 0 
1 1 1 1 0 1 0 0 0 
1 0 0 0 0 0 1 1 0 

lac.out

Datele de intrare corespund restrictiilor impuse

2 4

Rezolvare

<syntaxhighlight lang="python" line="1" start="1">

def validare(nr_n, nr_m, matrix): # functia de validare a datelor de intrare

   if nr_n < 1 or nr_n > 1000 or nr_m < 1 or nr_m > 1000:
       raise ValueError
   for linie in matrix:
       if len(linie) != nr_m:
           raise ValueError
   file_out.write("Datele de intrare corespund restrictiilor impuse\n")

def dfs(i, j, nr_n, nr_m, matrix, vizitat):

   dx = [-1, 0, 1, 0]
   dy = [0, 1, 0, -1]
   vizitat[i][j] = True
   for directie in range(4):
       x_nou, y_nou = i + dx[directie], j + dy[directie]
       if 0 <= x_nou < nr_n and 0 <= y_nou < nr_m and matrix[x_nou][y_nou] == 1 and not vizitat[x_nou][y_nou]:
           dfs(x_nou, y_nou, n, m, matrix, vizitat)

def numara_insule_peninsule(nr_n, nr_m, matrix): # functia de rezolvare

   vizitat = [[False]*nr_m for _ in range(nr_n)]
   insule = peninsule = 0
   for i in range(nr_n):
       for j in range(nr_m):
           if matrix[i][j] == 1 and not vizitat[i][j]:
               if i == 0 or i == nr_n-1 or j == 0 or j == nr_m-1:
                   peninsule += 1
               else:
                   insule += 1
               dfs(i, j, nr_n, nr_m, matrix, vizitat)
   file_out.write(f'{insule} {peninsule}\n')

if __name__ == '__main__':

   try:
       file_in = open('lacin.txt', 'r')
       file_out = open('lacout.txt', 'w')
       n, m = map(int, file_in.readline().split())
       matrice = [list(map(int, file_in.readline().split())) for _ in range(n)]
       validare(n, m, matrice)
       numara_insule_peninsule(n, m, matrice)
   except ValueError:
       file_out.write("Datele de intrare nu corespund restrictiilor impuse\n")
   except IndexError:
       file_out.write("Datele de intrare nu corespund restrictiilor impuse\n")

</syntaxhighlight>