Задать двумерный массив. Посчитать сумму элементов главной диагонали

# include <iostream.h>      //подключение библиотеки
void main (){               //главная функция программы
const n=5;
int mas [n] [n];            //двумерный массив n на n
for (int i = 0; i < n; i++)
for (int j = 0; j < n; j++)
cin >> mas [i] [j];         //ввод массива с клавиатуры
int S (0);
for (int i = 0; i < n; i++)
S+=mas [i] [i];         //подсчет суммы элементов главной диагонали
cout << S << endl;          //вывод результата на экран
cin.get ();
cin.get();
}

# include <iostream>      //подключение библиотеки
using namespace std;
 
int main ()
{               //главная функция программы
const int n=5;
int mas [n] [n];            //двумерный массив n на n
for (int i = 0; i < n; i++)
for (int j = 0; j < n; j++)
cin >> mas [i] [j];         //ввод массива с клавиатуры
int S (0);
for (int i = 0; i < n; i++)
S+=mas [i] [i];         //подсчет суммы элементов главной диагонали
cout << S << endl;          //вывод результата на экран
cin.get ();
cin.get();
return 0;
}