Editing 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

# 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)]

# 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)

# 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.