Editing 1383 - Avioane (section)

=== Explicație ===
În matricea de <code>8x8</code> avem <code>2</code> avioane, ambele de tip <code>3</code>. Cabinele sunt în coordonatele <code>(5,6)</code> și <code>(8,4)</code>. Deci pe prima linie în fișierul de ieșire va fi <code>2</code>, iar pe următoarele <code>4</code> linii numărul avioanelor de fiecare tip: <code>0</code>, <code>0</code>, <code>2</code>, <code>0</code>, fiecare pe linie nouă.

În această reprezentare avem <code>5</code> lovituri. Cu cele <code>5</code> lovituri am reușit să lovim ambele avioane, fără a lovi cabina sau cel puțin <code>5</code> lovituri/avion. Astfel următoarele <code>2</code> valori din fișierul de ieșire sunt <code>2</code> și <code>0</code>.

În această diagramă nu mai putem adăuga un avion de tip <code>1</code> astfel pe ultima linie vom avea valoarea <code>0</code>.

<syntaxhighlight lang="python" line="1">
def process_game(N, M, avioane, L, lovituri):
    # Initializare matrice
    matrice = [[0] * N for _ in range(N)]
    
    # Plasarea avioanelor în matrice
    avion_celule = {}
    avion_tip = {}
    
    for avion in avioane:
        id, tip, linie, coloana = avion
        avion_celule[id] = []
        avion_tip[id] = tip
        
        if tip == 1:  # cabina spre nord
            coords = [(linie, coloana), (linie+1, coloana-2), (linie+1, coloana-1), (linie+1, coloana),
                      (linie+1, coloana+1), (linie+1, coloana+2), (linie+2, coloana), 
                      (linie+3, coloana-1), (linie+3, coloana), (linie+3, coloana+1)]
        elif tip == 2:  # cabina spre est
            coords = [(linie, coloana), (linie-2, coloana-1), (linie-1, coloana-1), (linie, coloana-1),
                      (linie+1, coloana-1), (linie+2, coloana-1), (linie, coloana-2), 
                      (linie-1, coloana-3), (linie, coloana-3), (linie+1, coloana-3)]
        elif tip == 3:  # cabina spre sud
            coords = [(linie, coloana), (linie-1, coloana-2), (linie-1, coloana-1), (linie-1, coloana),
                      (linie-1, coloana+1), (linie-1, coloana+2), (linie-2, coloana),
                      (linie-3, coloana-1), (linie-3, coloana), (linie-3, coloana+1)]
        elif tip == 4:  # cabina spre vest
            coords = [(linie, coloana), (linie-2, coloana+1), (linie-1, coloana+1), (linie, coloana+1),
                      (linie+1, coloana+1), (linie+2, coloana+1), (linie, coloana+2),
                      (linie-1, coloana+3), (linie, coloana+3), (linie+1, coloana+3)]
        
        for coord in coords:
            r, c = coord
            if 0 <= r < N and 0 <= c < N:
                matrice[r][c] = id
                avion_celule[id].append((r, c))
    
    # Procesarea loviturilor
    lovituri_set = set(lovituri)
    
    avioane_doborate = 0
    avioane_avariate = 0
    
    for id, celule in avion_celule.items():
        lovituri_in_celule = [celula for celula in celule if celula in lovituri_set]
        numar_lovituri = len(lovituri_in_celule)
        cabina_lovita = celule[0] in lovituri_set
        
        if numar_lovituri >= 5 or cabina_lovita:
            avioane_doborate += 1
        elif numar_lovituri > 0:
            avioane_avariate += 1
    
    # Numărarea tipurilor de avioane
    tipuri_avioane = {1: 0, 2: 0, 3: 0, 4: 0}
    for tip in avion_tip.values():
        tipuri_avioane[tip] += 1
    
    return M, tipuri_avioane, avioane_avariate, avioane_doborate

# Exemplu de utilizare
N = 10
M = 2
avioane = [
    (1, 1, 2, 2),  # avion 1, tip 1 (nord), cabina la (2, 2)
    (2, 2, 5, 5)   # avion 2, tip 2 (est), cabina la (5, 5)
]
L = 3
lovituri = [(2, 2), (5, 5), (4, 5)]

total_avioane, tipuri_avioane, avioane_avariate, avioane_doborate = process_game(N, M, avioane, L, lovituri)
print(f"Numărul total de avioane: {total_avioane}")
print(f"Numărul de avioane de fiecare tip: {tipuri_avioane}")
print(f"Numărul de avioane avariate, fără a fi doborâte: {avioane_avariate}")
print(f"Numărul de avioane doborâte: {avioane_doborate}")
</syntaxhighlight>