0220 - Prod Max

Sursa: [1]

Enunt

Se consideră tabloul bidimensional cu m linii şi n coloane, care conţine doar valorile {0,1,2}.

Cerinţa

Să se determine numerele de ordine ale coloanelor pentru care produsul elementelor este maxim.

Date de intrare

Fişierul de intrare prodmax.in conţine pe prima linie numerele m şi n, iar pe următoarele m linii câte n numere naturale separate prin spaţii, reprezentând elementele tabloului.

Date de ieșire

Pe ecran se va afișa mesajul: "Datele de intrare corespund restricțiilor impuse.".Fişierul de ieşire prodmax.out va conţine pe prima linie numerele cerute, separate prin spatii, în ordine crescătoare. În cazul în care datele introduse de la tastatură nu îndeplinesc cerințele enunțate, programul va afișa "Datele de intrare nu corespund restricțiilor impuse.".

Restricţii şi precizări

• 2≤m,n≤100
• numere de ordine ale liniilor sunt între 1 şi m
• numere de ordine ale coloanelor sunt între 1 şi n

Exemple

Exemplu 1

prodmax.in

4 4

2 1 1 0

1 1 1 1

2 2 2 1

1 2 1 1

prodmax.out

1 2

Rezolvare

<syntaxhighlight lang="python" line> def citire_lin_col():

   with open("prodmax.in", "r") as fin:
       try:
           n, m = map(int, fin.readline().split())
           if 2 <= n <= 100 and 2 <= m <= 100:
               print("Datele sunt corecte.")
               return n, m
           else:
               print("Datele nu sunt conform restrictiilor impuse.")
               exit()
       except ValueError:
           print("Trebuie introduse doar numere intregi.")
           exit()

def citeste_valori_matrice(m, n):

   matrice = []
   with open("prodmax.in", "r") as fin:
       fin.readline()
       for linie in range(m):
           matrice.append(list(map(int, fin.readline().split())))
           if len(matrice[linie]) != n:
               print("Dimensiunile matricei nu sunt cele așteptate.")
               exit()
           for elem in matrice[linie]:
               if elem not in [0, 1, 2]:
                   print("Elementele matricei nu sunt corecte.")
                   exit()
   return matrice

def prodmax():

   n, m = citire_lin_col()
   a = citeste_valori_matrice(m, n)
   pmax = -1
   for j in range(m):
       p = 1
       for i in range(n):
           p *= a[i][j]
       if p > pmax:
           pmax = p
   with open("prodmax.out", "w") as fout:
       for j in range(m):
           p = 1
           for i in range(n):
               p *= a[i][j]
           if p == pmax:
               fout.write(str(j + 1) + " ")

if _name_ == "_main_":

   prodmax()

</syntaxhighlight>

Explicații

Acest cod implementează o funcție numită prodmax() care citește o matrice de dimensiune n x m și găsește numărul de coloană a produsului maxim de elemente din acea matrice.

Funcția citire_lin_col() este utilizată pentru a citi numerele întregi n și m din fișierul de intrare prodmax.in și verifica dacă sunt în conformitate cu restricțiile impuse (2 <= n <= 100 și 2 <= m <= 100).

Funcția citeste_valori_matrice(m, n) este utilizată pentru a citi valorile matricei din fișierul de intrare prodmax.in. În timpul citirii, se verifică dacă dimensiunile matricei sunt cele așteptate și dacă elementele matricei sunt corecte (adica 0, 1 sau 2).

În funcția principală prodmax(), matricea este citită prin apelarea funcțiilor citire_lin_col() și citeste_valori_matrice(m, n). Apoi, se găsește produsul maxim de elemente din fiecare coloană, iar coloana cu cel mai mare produs este înregistrată în variabila pmax. Această coloană este găsită prin parcurgerea matricei și înmulțind elementele de pe fiecare coloană, apoi comparând produsul cu pmax. Dacă produsul este egal cu pmax, atunci numărul de coloană este scris în fișierul de ieșire prodmax.out.

În general, acest cod utilizează citirea din fișier și scrierea în fișier pentru a efectua o operație matriceală simplă și returnează numărul de coloană cu cel mai mare produs de elemente.