Vizualizare sursă pentru „0991 - Matrice 5”

= Cerința =
Se dă o matrice pătratică cu <code>n</code> linii și <code>n</code> coloane, numerotate de la <code>1</code> la <code>n</code> și elemente numere naturale. Să se determine suma elementelor aflate strict deasupra diagonalei secundare a matricei.

Se va defini și folosi subprogramul <code>sub</code>, cu <code>3</code> parametri:

* <code>n</code> – dimensiunea matricei
* <code>x</code> – matricea
* <code>k</code> – un număr natural, <code>1 < k ≤ 2*n</code>

care va returna suma tuturor elementelor <code>aij</code> din matrice pentru care <code>i + j = k</code>.

!!!Variabila k nu va fi folosită deloc. Programul va afișa doar suma elementelor de pe diagonala secundară.

= Date de intrare =
Fișierul de intrare <code>matriceIN.txt</code> conține pe prima linie numărul <code>n</code>, iar următoarele <code>n</code> linii câte <code>n</code> numere naturale, reprezentând elementele matricei.

= Date de ieșire =
Fișierul de ieșire <code>matriceOUT.txt</code> va conține pe prima linie suma determinată. În cazul în care restricțiile nu sunt îndeplinite, se va afișa un mesaj corespunzător.

= Restricții și precizări =

* <code>1 ≤ n ≤ 100</code>
* elementele matricei sunt numere naturale mai mici decât <code>1000</code>
* liniile matricei sunt numerotate de la <code>1</code> la <code>n</code>.

== Exemplul 1 ==
Input:

4

1 6 3 1

6 1 3 1

1 3 1 6

3 1 6 1

Output:

18

== Exemplul 2 ==
Input:

121

1 6 3 1

6 1 3 1

1 3 1 6

3 1 6 1

Output consola:

n trebuie să fie între 1 și 100 inclusiv.

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def sub(n, x, k=0):
    if not check_constraints(n,matrix):
        return
    suma = 0
    for i in range(0, n-1):
        for j in range(0, n-1-i):
                suma += x[i][j]
    return suma
def check_constraints(n, matrix):
    if not(1 <= n <= 100):
        print("n trebuie să fie între 1 și 100 inclusiv.") 
    for row in matrix:
        for elem in row:
            if elem >= 1000:
                print("Elementele matricei trebuie să fie mai mici decât 1000.")
# Citirea datelor de intrare
with open("matriceIN.txt", "r") as fin:
    n = int(fin.readline())
    matrix = [list(map(int, fin.readline().split())) for _ in range(n)]
# Calculul si afisarea sumei elementelor deasupra diagonalei secundare in fisierul de iesire
with open("matriceOUT.txt", "w") as fout:
    fout.write(str(sub(n,matrix)))

</syntaxhighlight>