Vizualizare sursă pentru „0231 - Exista”

Sursă: [https://www.pbinfo.ro/probleme/231/exista]
== Cerință ==
Scrieţi un program care citeşte un tablou bidimensional cu '''m''' linii şi '''n''' coloane şi un număr '''k''' şi determină suma indicilor coloanelor care conţin valoarea '''k'''.
== Date de intrare ==
Fişierul de intrare '''exista.in''' conţine pe prima linie numerele '''k, m''' şi '''n''', iar pe următoarele '''m''' linii câte '''nv numere naturale separate prin spaţii, reprezentând elementele tabloului.
== Date de ieșire == 
Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt corecte.". Fişierul de ieşire '''exista.out''' va conţine pe prima linie suma cerută. Dacă nu există în matrice valoarea '''k''', se va afişa mesajul '''NU EXISTA'''. În caz contrar, se va afișa pe ecran: "Datele nu sunt comform restricțiilor impuse.".
== Restricții și precizări ==
* '''2≤m,n≤10''';
* valoarea lui '''k''' şi elementele matricei sunt numere întregi cu cel mult '''4''' cifre;
* liniile sunt numerotate de la '''1''' la '''m''', iar coloanele de la '''1''' la '''n'''.
== Exemple ==
=== Exemplu 1 ===
; '''exista.in'''
: 3 5 4
: 2 4 5 -8
: 1 3 7 9
:  -2 3 10
: 5 4 2 37
: 6 7 3 13
; '''exista.in'''
: 5
==== Explicatie ====
Coloana '''2''' şi coloana '''3''' conţin valoarea '''3''', iar suma lor este '''5'''.
== Rezolvare ==
<syntaxhighlight lang="python" line="1">
def citire_matrice():
    # Deschide fișierul de intrare și citește k, m și n
    with open("exista.in", "r") as fin:
        line = fin.readline().split()
        if len(line) != 3:
            print("Datele nu sunt conform restrictiilor impuse.")
            exit()
        k = float(line[0])
        m, n = map(int, line[1:])
        if not (2 <= m <= 10 and 2 <= n <= 10):
            print("Datele nu sunt conform restrictiilor impuse.")
            exit()
        matrice = []
        # Citește matricea din fișierul de intrare
        for i in range(m):
            row = list(map(float, fin.readline().split()))
            # Verifică dacă elementele matricei sunt conform cerințelor
            if not all(isinstance(x, float) for x in row):
                print("Datele nu sunt conform restrictiilor impuse.")
                exit()
            matrice.append(row)
        return k, m, n, matrice


def calculeaza_suma(k, m, n, matrice):
    suma = 0.0
    coloane_cu_k = []
    for j in range(n):
        coloana_j = [matrice[i][j] for i in range(m)]
        if k in coloana_j:
            suma += j + 1
            coloane_cu_k.append(j + 1)
    if not coloane_cu_k:
        print("NU EXISTA")
        exit()
    return suma


if _name_ == "_main_":
    k, m, n, matrice = citire_matrice()
    suma = calculeaza_suma(k, m, n, matrice)
    with open("exista.out", "w") as fout:
        fout.write(str(suma) + "\n")
    print("Suma indicilor coloanelor care contin valoarea k este:", suma)
</syntaxhighlight>
== Explicatie ==
 Acest cod rezolvă problema de a calcula suma indicilor coloanelor care conțin o anumită valoare k într-o matrice bidimensională dată.

 Funcția citire_matrice deschide fișierul de intrare exista.in și citește valoarea k, dimensiunile matricei (m și n), și matricea în sine. Dacă datele de intrare nu respectă condițiile impuse (de exemplu, dimensiunile matricei sunt în afara intervalului [2,10]), se afișează un mesaj de eroare și programul se încheie cu exit(). Funcția returnează k, m, n și matricea.

 Funcția calculeaza_suma primește k, m, n și matricea, și calculează suma indicilor coloanelor care conțin valoarea k. Pentru fiecare coloană, extrage toate elementele acelei coloane și verifică dacă k se află printre acele elemente. Dacă da, adaugă indicele coloanei (indexul începe de la 1, nu de la 0) la lista coloane_cu_k și adaugă indicele la variabila suma. Dacă nu există nicio coloană care să conțină valoarea k, afișează mesajul "NU EXISTA" și programul se încheie cu exit(). Funcția returnează suma.

 În blocul principal, se apelează funcțiile citire_matrice și calculeaza_suma pentru a obține suma și se scrie aceasta în fișierul de ieșire exista.out. Se afișează, de asemenea, valoarea sumei în consolă.