Diagonala principală într-o matrice pătratică

O matrice pătratică are același număr de linii și coloane: n. Diagonala principală conține celulele cu indicele liniei egal cu indicele coloanei: a[0][0], a[1][1], ... a[n-1][n-1]. Exemplu: pentru matricea cu liniile 1 2 și 3 4, diagonala este 1 și 4, deci suma este 5. În C++: long long suma = 0; for (int i = 0; i < n; i++) suma += a[i][i];. Pentru n=1, diagonala are un singur element. Nu aplica aceeași regulă fără să verifici cerința unei matrici dreptunghiulare.