2278 - MMultiplication

From Bitnami MediaWiki
Revision as of 10:02, 12 December 2023 by Mesarosdenisa (talk | contribs)

Cerinta

Se dau două matrice cu elemente numere întregi. Determinați produsul lor, dacă este posibil.

Date de intrare

Fișierul de intrare mmultiplication.txt conține pe prima linie numerele n și m reprezentând numărul de linii, respectiv de coloane, al matricii A. Pe următoarele n linii sunt câte m numere întregi separate printr-un spațiu, reprezentând elementele matricii A. Pe linia n+2 a fișierului se află două numere p și q reprezentând numărul de linii, respectiv de coloane, al matricii B. Pe următoarele p linii sunt câte q numere întregi separate printr-un spațiu, reprezentând elementele matricii B.

Date de iesire

Fișierul de ieșire mmultiplication.txt va conține pe prima linie numărul -1, în cazul în care matricele nu pot fi înmulțite. Dacă matricele pot fi înmulțite, se va afișa matricea rezultată după înmulțirea celor două, câte o linie a matricei pe o linie din fișier, elementele fiecărei linii fiind separate prin câte un spațiu.

Restrictii si precizari

  • 1 ≤ n, m, p, q ≤ 1000
  • elementele matricelor vor fi numere întregi cu valoarea absolută mai mică decât 2000

Exemplul 1

Intrare
4 3
1 2 3
4 5 6
7 8 9
10 11 12
3 5
1 2 3 4 5
6 7 8 9 10
11 12 13 14 0
Iesire
Datele introduse corespund restrictiilor impuse
46 52 58 64 25
100 115 130 145 70
154 178 202 226 115
208 241 274 307 160

Exemplul 2

Intrare
2 3
3600 0 -10
54 2479 1
3 2
0 0
1 1
0 1
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 inmulteste_matrici(matrice_A, n_A, m_A, matrice_B, n_B, m_B):

   if m_A != n_B:
       raise ValueError("Numarul de coloane al primei matrici nu coincide cu numarul de linii al celei de-a doua matrici")
   matrice_produs = [[0] * m_B for _ in range(n_A)]
   for i in range(n_A):
       for j in range(m_B):
           for k in range(m_A):
               matrice_produs[i][j] += matrice_A[i][k] * matrice_B[k][j]
   return matrice_produs, n_A, m_B

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("mmultiplication.txt", "r") as f:
       matrice_A, n_A, m_A = citeste_matrice(f)
       matrice_B, n_B, m_B = citeste_matrice(f)
   # Calcularea produsului matricelor
   try:
       matrice_produs, n_produs, m_produs = inmulteste_matrici(matrice_A, n_A, m_A, matrice_B, n_B, m_B)
       # Scrierea rezultatului in fisierul de iesire
       with open("mmultiplication.txt", "w") as g:
           scrie_matrice(g, matrice_produs, n_produs, m_produs)
   except ValueError as e:
       # Nu se poate efectua inmultirea
       with open("mmultiplication.txt", "w") as g:
           g.write("-1")

if __name__ == "__main__":

   main()

</syntaxhighlight>