Editing 1395 - MSuma

== Cerinta ==

Se dau două matrice cu elemente numere întregi. Calculați suma lor.

== Date de intrare ==

Fișierul de intrare msuma.txt conține pe prima linie numerele n m, reprezentând dimensiunile primei matrice. Urmează n linii cu câte m valori, elementele primei matrice.

Linia n+2 conține numerele p q, reprezentând dimensiunile celei de-a doua matrice. Urmează p linii cu câte q valori, elementele celei de-a doua matrice.

== Date de iesire ==

Fișierul de ieșire msuma.txt va conține pe prima linie numerele S T, reprezentând dimensiunile matricei sumă. Următoarele S linii vor conține câte T valori separate prin exact un spațiu, reprezentând valorile din matricea sumă.

== Restrictii si precizari ==

*1 ≤ n , m , p , q ≤ 100;
*valoarea absolută a elementelor fiecărei matrice vor fi mai mici decât 1.000.000.000;
*adunarea matricelor se poate efectua pentru fiecare test.

== Exemplul 1 ==
;Intrare
;3 4
;2 1 4 6 
;5 -8 5 5 
;2 5 -2 4 
;3 4
;3 7 -5 0 
;1 -4 1 5 
;-4 2 8 7 

;Iesire
;Datele introduse corespund restrictiilor impuse
;3 4
;5 8 -1 6 
;6 -12 6 10 
;-2 7 6 11 

== Exemplul 2 ==
;Intrare
;2 3
;-3 7 0
;8000 1 -2
;3 2
;7 0
;5438 870 
;0 0

;Iesire
;Datele introduse nu corespund restrictiilor impuse


== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def citeste_matrice(fisier):
    # Citeste dimensiunile matricei
    n, m = map(int, fisier.readline().split())
    matrice = []
    
    # Citeste elementele matricei
    for _ in range(n):
        linie = list(map(int, fisier.readline().split()))
        matrice.append(linie)

    return matrice, n, m

def suma_matricelor(matrice1, n1, m1, matrice2, n2, m2):
    if n1 != n2 or m1 != m2:
        raise ValueError("Dimensiunile matricelor nu corespund")

    matrice_suma = [[0] * m1 for _ in range(n1)]

    for i in range(n1):
        for j in range(m1):
            matrice_suma[i][j] = matrice1[i][j] + matrice2[i][j]

    return matrice_suma, n1, m1

def scrie_matrice(fisier, matrice, n, m):
    fisier.write(f"{n} {m}\n")
    for linie in matrice:
        fisier.write(" ".join(map(str, linie)) + "\n")

def main():
    # Citirea datelor de intrare din fisier
    with open("msuma.in", "r") as f:
        matrice1, n1, m1 = citeste_matrice(f)
        matrice2, n2, m2 = citeste_matrice(f)

    # Calcularea sumei matricelor
    try:
        matrice_suma, n_suma, m_suma = suma_matricelor(matrice1, n1, m1, matrice2, n2, m2)

        # Scrierea rezultatului în fisierul de iesire
        with open("msuma.txt", "w") as g:
            scrie_matrice(g, matrice_suma, n_suma, m_suma)

    except ValueError as e:
        print(f"EROARE: {e}")

if __name__ == "__main__":
    main()
</syntaxhighlight>