0783 - Diagonale 1

Cerinţa

Se dă o matrice cu n linii şi n coloane şi elemente numere naturale. Să se determine suma elementelor de pe cele două diagonale vecine cu diagonala principală.

Date de intrare

Programul citește de la tastatură numărul n, iar apoi n*n numere naturale, separate prin spaţii, reprezentând elementele matricei, linie cu linie.

Date de ieşire

Dacă datele sunt introduse corect,pe ecran se va afișa :"Datele sunt introduse corect.",apoi pe un rând nou numărul S, reprezentând valoarea calculată.În cazul contrar,se va afișa pe ecran "Datele nu corespund restricțiilor impuse.".

Restricții și precizări

• 1 ⩽ n ⩽ 100
• elementele matricei vor fi mai mici decât 1.000.000

Exemplu

Date de intrare

5

3 1 8 5 4

7 8 5 1 2

2 2 6 7 3

9 8 1 3 6

7 5 3 1 7

Date de ieșire

Datele sunt introduse corect.

30

Rezolvare

def validare_date(n, a):
    # Verifică valoarea lui n
    if not (1 <= n <= 100):
        print("Datele nu corespund restricțiilor impuse.")
        return False

    # Verifică valorile elementelor matricei
    for i in range(n):
        for j in range(n):
            if not (0 <= a[i][j] < 1000000):
                print("Datele nu corespund restricțiilor impuse.")
                return False

    # Datele sunt introduse corect
    return True
#Funcția care calculează suma diagonalelor
def calculeaza_suma_diagonalelor(n, a):
    s = 0
    for i in range(1, n):
        s += a[i][i-1] + a[i-1][i]
    return s


if __name__ == '__main__':
    # Citire date de intrare
    n = int(input())
    a = [[int(x) for x in input().split()] for i in range(n)]

    # Verificare date de intrare
    if validare_date(n, a):
        print("\nDatele sunt introduse corect.\n")
        # Calculare si afișare rezultat
        S = calculeaza_suma_diagonalelor(n, a)
        print(S)
    else:
        print("Datele nu corespund restricțiilor impuse.")