0234 - Med Poz

Enunț

Se consideră un tablou bidimensional cu n linii şi n coloane ce conţine numere intregi din intervalul [-100,100].

Cerinţa

Să se determine media aritmetică a elementelor strict pozitive din matrice, care sunt situate sub diagonala principală.

Date de intrare

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

Date de ieşire

Dacă datele sunt introduse corect,în fişierul de ieşire medpoz.out se va afișa :"Datele sunt introduse corect.",apoi pe un rând va conţine un singur număr, reprezentând valoarea cerută.În cazul contrar,se va afișa pe ecran "Datele nu corespund restricțiilor impuse.".

Restricții și precizări

2 ⩽ n ⩽ 20
dacă în matrice nu există elemente strict pozitive, situate sub diagonala principală, se va afișa mesajul NU EXISTA;
media se va afişa cu exact trei zecimale, fără rotunjire;

Exemplu

medpoz.in: 4; -1 2 4 5; 0 6 3 1; 2 4 2 0; 3 -5 1 -3
medpoz.out: Datele sunt introduse corect.; 2.500

Explicație

Valorile strict pozitive situate sub diagonala principală sunt 2, 4, 3 şi 1, iar media lor aritmetică este 2.5. Se va afişa cu trei zecimale, adică 2.500

Rezolvare

def validare_date(n: int, a: list[list[int]]) -> bool:

   # Verificare dimensiunea matricei
   if not (2 <= n <= 20):
       return False
   # Verificare interval valori elemente
   for i in range(n):
       for j in range(n):
           if not (-100 <= a[i][j] <= 100):
               return False
   return True

def media_aritmetica_pozitive_sub_diagonala(n: int, a: list[list[int]]) -> float:

   suma_pozitive_sub_diagonala = 0
   numar_elemente_pozitive_sub_diagonala = 0
   # Parcurgere elemente sub diagonala principală
   for i in range(1, n):
       for j in range(i):
           # Dacă elementul este strict pozitiv, se adună la suma S
           if a[i][j] > 0:
               suma_pozitive_sub_diagonala += a[i][j]
               numar_elemente_pozitive_sub_diagonala += 1
   # Daca nu s-au găsit elemente pozitive sub diagonală, se returnează None
   if numar_elemente_pozitive_sub_diagonala == 0:
       return None
   # Altfel, se calculează și se returnează media aritmetică
   else:
       return suma_pozitive_sub_diagonala / numar_elemente_pozitive_sub_diagonala

if __name__ == '__main__':

   # Citire date de intrare
   with open('medpoz.in', 'r') as fin:
       n = int(fin.readline())
       a = []
       for i in range(n):
           linie = list(map(int, fin.readline().split()))
           a.append(linie)
   # Verificare date de intrare
   if not validare_date(n, a):
       with open('medpoz.out', 'w') as fout:
           fout.write("Datele nu corespund restricțiilor impuse.")
   else:
       # Calcul și afișare rezultat
       rezultat = media_aritmetica_pozitive_sub_diagonala(n, a)
       if rezultat is None:
           print("NU EXISTA")
       else:
           with open('medpoz.out', 'w') as fout:
               fout.write("Datele sunt introduse corect.\n")
               fout.write(f"{rezultat:.3f}")

</syntaxhighlight>