Editing 1800 - Matop

==Enunt==
Ilie a fost chemat de urgenţă în localitatea natală, la Fărcășești, pentru a rezolva o problemă care ar ajuta la reducerea poluării provocate de mina din localitate. Din păcate soluţia lui Ilie rezolvă optim doar un sfert din problemă aşa că vă cere ajutorul. Problema este următoarea:

Se dă o matrice pătratică cu N linii şi N coloane care iniţial are toate elementele egale cu 0. Pe această matrice se execută 4 tipuri de operaţii:

:1 LIN VAL: toate elementele de pe linia LIN cu valoarea mai mică decât VAL iau valoarea VAL.
:2 COL VAL: toate elementele de pe coloana COL cu valoarea mai mică decât VAL iau valoarea VAL.
:3 LIN COL: să se afişeze valoarea elementului de pe linia LIN şi coloana COL.
:4: să se afişeze suma elementelor de pe diagonala principală.

==Cerinţa==
Dat fiind N şi K operaţii, să se execute fiecare în ordinea citirii.

==Date de intrare==
Fişierul de intrare matop.in va conţine pe prima linie două numere naturale N, K reprezentând dimensiunile matricei şi numărul de operaţii care trebuie efectuate. Pe următoarele K linii se află descrierea unei operaţii în formatul precizat mai sus.

==Date de ieşire==
Fişierul matop.out va conţine pentru fiecare operaţie de tip 3 şi 4 din fişierul de intrare răspunsul opeaţiei corespunzătoare.

==Restricţii şi precizări==
1 ≤ N ≤ 25.000
1 ≤ K ≤ 100.000
1 ≤ LIN ≤ N
1 ≤ COL ≤ N
0 ≤ VAL ≤ 100.000
Pentru 40% din teste 1 ≤ N, K ≤ 100

==Exemplu==
;matop.in
:5 11
:1 1 1
:2 2 3
:3 1 4
:2 3 9
:1 5 2
:1 4 4
:3 4 3
:3 1 2
:1 1 10
:3 1 3
:4

;matop.out
:1
:9
:3
:10
:28

==Rezolvare==
<syntaxhighlight lang="python" line>
#1800 matop
def execute_operations(N, K, operations):
    matrix = [[0] * N for _ in range(N)]
    diagonal_sum = 0
    output = []

    for op in operations:
        op_type, *args = op

        if op_type == 1:
            lin, val = args
            for j in range(N):
                matrix[lin - 1][j] = max(matrix[lin - 1][j], val)
        elif op_type == 2:
            col, val = args
            for i in range(N):
                matrix[i][col - 1] = max(matrix[i][col - 1], val)
        elif op_type == 3:
            lin, col = args
            output.append(matrix[lin - 1][col - 1])
        elif op_type == 4:
            diagonal_sum += sum(matrix[i][i] for i in range(N))

    if diagonal_sum:
        output.append(diagonal_sum)

    # Scrierea rezultatelor în fișierul 'matop.out'
    with open('matop.out', 'w') as file_out:
        for result in output:
            file_out.write(str(result) + '\n')

# Citirea datelor de intrare
with open('matop.in', 'r') as file:
    N, K = map(int, file.readline().split())
    operations = [list(map(int, line.split())) for line in file.readlines()]

# Executarea operațiilor și afișarea rezultatelor în fișierul 'matop.out'
execute_operations(N, K, operations)
</syntaxhighlight>