4050 - amongus

Enunt

În jocul Among Us un echipaj încearcă să piloteze o navetă spațială. O parte din ei încearcă să saboteze misiunea. Acești „impostori” sunt de obicei periculoși și astfel astronauții (sau „crewmates”) trebuie să se ferească pe cât posibil de aceștia. Astronauții au studiat planul interiorului navetei și au constatat că este compartimentat în nxm cabine, dispuse pe n rânduri, câte m pe fiecare rând, iar din fiecare cabină se poate trece în oricare altă cabină alăturată situată într-una din cele 8 direcții (N, E, S, V, N-V, N-E, S-V, S-E). Astronautii au creat o matrice a codurilor asociind fiecărei cabine un număr din mulțimea {0,1,2} cu semnificația:

• cod 0 dacă este o cabină sigură (adică în cabină nu se află niciun impostor)
• cod 1 dacă în ea se află un impostor
• cod 2 dacă este un portal („vent”).

Cerinta

Cunoscând matricea codurilor, pentru a ajuta astronauții, vi se cere:

• 1) să determinați numărul total de cabine sigure, nu sunt neapărat accesibile, ce nu conțin portale
• 2) să marcați în matricea codurilor cu codul 3 fiecare cabină sigură (și accesibilă)
• 3) să determinați numărul portalelor situate în cabine accesibile.

Date de intrare

Fișierul de intrare amongus.txt conține:

• pe prima linie numărul p reprezentând cerința care se rezolvă (1, 2 sau 3)
• pe a doua linie, două numere naturale n și m cu semnificația din enunț
• pe fiecare din următoarele n linii, câte m numere naturale din mulțimea {0,1,2}, separate prin câte un spațiu, reprezentând matricea de codificare a interiorului navetei spațiale.

Date de iesire

Fișierul de ieșire amongus.out va conține:

• dacă p=1, pe prima linie, un număr natural reprezentând răspunsul la cerința 1
• dacă p=2, n linii cu câte m numere naturale, separate prin câte un spațiu,reprezentând matricea codurilor modificată conform cerinței 2
• dacă p=3, pe prima linie, un număr natural reprezentând răspunsul la cerința 3.

Restrictii si precizari

• 1 ≤ n, m ≤ 100
• astronauții pot intra în naveta doar prin cabina de coordonate (1,1) ceea ce înseamnă că accesul în unele cabine nu se poate realiza deși sunt sigure
• în cazul în care în prima cabină se află un impostor, pentru oricare dintre cerințe, se va scrie în fișier mesajul misiune esuata
• pentru cerința 1, se va modifica în 3 codul fiecărei cabine sigure
• un portal poate fi situat doar într-o cabină sigură
• există și cabine sigure care nu conțin portale
• impostorii nu se deplasează în afara cabinei în care se află inițial

Exemplul 1

Intrare

1

3 4

0 2 1 0

2 0 1 0

0 0 1 2

Iesire

Datele introduse corespund restrictiilor impuse

6

Exemplul 2

Intrare

8

-1 7

0 0 2 5

-8 6 1 0

0 0 0 -3

Iesire

Datele introduse nu corespund restrictiilor impuse

Rezolvare

<syntaxhighlight lang="python3" line="1"> def este_valid(i, j, n, m):

   return 0 <= i < n and 0 <= j < m

def dfs(matrice, i, j, n, m):

   if not este_valid(i, j, n, m) or matrice[i][j] != 0:
       return

   matrice[i][j] = 3

   for dx in [-1, 0, 1]:
       for dy in [-1, 0, 1]:
           if dx == 0 and dy == 0:
               continue
           ni, nj = i + dx, j + dy
           dfs(matrice, ni, nj, n, m)

def cerinta_1(matrice, n, m):

   count = 0
   for i in range(n):
       for j in range(m):
           if matrice[i][j] == 0:
               count += 1
   return count

def cerinta_2(matrice, n, m):

   matrice_modificata = [row.copy() for row in matrice]
   for i in range(n):
       for j in range(m):
           if matrice[i][j] == 0:
               dfs(matrice_modificata, i, j, n, m)
   return matrice_modificata

def cerinta_3(matrice, n, m):

   count = 0
   for i in range(n):
       for j in range(m):
           if matrice[i][j] == 2:
               count += 1
   return count

1. Citire date de intrare

with open("amongus.txt", "r") as fin:

   p = int(fin.readline().strip())
   n, m = map(int, fin.readline().split())
   matrice = [list(map(int, fin.readline().split())) for _ in range(n)]

1. Rezolvare cerință

if p == 1:

   rezultat = cerinta_1(matrice, n, m)

elif p == 2:

   rezultat = cerinta_2(matrice, n, m)

elif p == 3:

   rezultat = cerinta_3(matrice, n, m)

1. Scriere rezultat în fișier

with open("amongus.txt", "w") as fout:

   if p == 1 or p == 3:
       fout.write(str(rezultat))
   elif p == 2:
       for row in rezultat:
           fout.write(" ".join(map(str, row)) + "\n")

</syntaxhighlight>

Explicatie

Se rezolva cerința 1. Matricea codurilor conține 6 camere sigure, codificate cu 0.