Sa se scrie un program care afișează suma elementelor dintr-o matrice de pe diagonala principala

#include <iostream>  

using namespace std;  

int main()

{

   int n, a[100][100], i , j;

   cin>>n;

   for(i=1;i<=n;i++)

       for(j=1;j<=n;j++)

           cin>>a[i][j];

   //elementele de pe diagonala principala au indicii egali (adica i=j)

   int suma=0;

   for(i=1;i<=n;i++)

       suma=suma+a[i][i];

   cout<<suma;

   return 0;

}