View source for 0192 - Sum Col Max

Sursă: [https://www.pbinfo.ro/probleme/192/sumcolmax]
== 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.ou'''t 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ă.