0192 - Sum Col Max

Sursă: [1]

Enunt

Se dă o matrice cu n linii și m coloane și elemente numere întregi.

Cerinţa

Să se afișeze elementele de pe coloana din matrice cu suma elementelor maximă

Date de intrare

Fişierul de intrare sumcolmax.in conţine pe prima linie numerele n și m, iar următoarele n linii câte m numere întregi, reprezentând elementele matricei.

Date de ieșire

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt corecte.". Fişierul de ieşire sumcolmax.out va conţine 1 linii, cu n valori, separate prin spații, reprezentând elementele de pe coloana cu suma elementelor maximă. În caz contrar, se va afișa pe ecran: "Datele nu sunt comform restricțiilor impuse.".

Restricții și precizări

• 1 ≤ n,m ≤ 25
• elementele matricei aparțin intervalului [-1000,1000]
• dacă sunt mai multe coloane cu suma elementelor maximă, se va considera coloana cu indicele mai mic
• elementele coloanei rezultat se afișează de sus în jos

Exemple

Exemplu 1

sumcolmax.in

4 5

-2 -9 7 9 -7

1 -7 -8 -1 4

8 2 5 7 2

8 -9 -7 10 -9

sumcolmax.out

9 -1 7 10

Rezolvare

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

   # Deschide fișierul de intrare și citește n și m
   with open("sumcolmax.in", "r") as fin:
       n, m = map(int, fin.readline().split())

       # Verifică dacă datele sunt conform cerințelor
       if not (1 <= n <= 25 and 1 <= m <= 25):
           print("Datele nu sunt conform restrictiilor impuse.")
           exit()

       matrice = []

       # Citește matricea din fișierul de intrare
       for i in range(n):
           row = list(map(int, fin.readline().split()))

           # Verifică dacă elementele matricei sunt conform cerințelor
           if not all(-1000 <= x <= 1000 for x in row):
               print("Datele nu sunt conform restrictiilor impuse.")
               exit()

           matrice.append(row)

       return n, m, matrice

def sumcolmax(n, m, matrice):

   smax = -100001
   ind = 0
   for j in range(m):
       s = 0
       for i in range(n):
           s += matrice[i][j]
       if s > smax:
           smax = s
           ind = j

   max_col = []
   for i in range(n):
       max_col.append(matrice[i][ind])
   return max_col

if _name_ == "_main_":

   n, m, matrice = read_matrix()
   max_col = sumcolmax(n, m, matrice)
   with open("sumcolmax.out", "w") as fout:
       fout.write(" ".join(map(str, max_col)))
   print("Elementele de pe coloana cu suma maxima sunt:", max_col)

</syntaxhighlight>

Explicații

Acest cod primește o matrice cu n linii și m coloane, precum și suma tuturor elementelor de pe fiecare coloană a matricei. Scopul codului este de a identifica coloana cu cea mai mare sumă și de a returna elementele de pe acea coloană într-o listă. În primul rând, funcția read_matrix deschide fișierul de intrare, citește n și m, și apoi citeste matricea din fișier. Dacă datele de intrare nu sunt conforme cu cerințele, codul va afișa un mesaj de eroare și se va opri. Funcția sumcolmax parcurge fiecare coloană și calculează suma tuturor elementelor de pe acea coloană. Dacă suma este mai mare decât suma maximă până în acel moment, atunci această coloană devine noua coloană cu suma maximă. Apoi, funcția creează o listă care conține elementele de pe coloana cu suma maximă și o returnează. În if _name_ == "_main_":, codul citește matricea cu read_matrix, calculează elementele de pe coloana cu suma maximă cu sumcolmax, scrie rezultatul în fișierul de ieșire sumcolmax.out, și afișează elementele de pe coloana cu suma maximă.