С++ Передать в функцию квадратную матрицу и вычислить сумму элементов на ее главной диагонали и вернуть эту сумму.

#include <iostream>
using namespace std;
const auto n = 5U;
long long trace(int matrix[n][n]) {
auto sum = 0LL;
for (auto i = 0U; i < n; ++i) sum += matrix[i][i];
return sum;
}
void show(int matrix[n][n]) {
for (auto i = 0U; i < n; ++i) {
for (auto j = 0U; j < n; ++j) cout << ' ' << matrix[i][j];
cout.put('\n');
}
}
int main() {
int matrix[n][n] = {
{ 2, 7, 8, 3, 6 },
{ 6, 9, 1, 4, 7 },
{ 5, 3, 8, 7, 9 },
{ 7, 1, 3, 9, 2 },
{ 2, 1, 0, 3, 4 }
};
show(matrix);
cout.put('\n');
auto sum = trace(matrix);
cout << " Sum: " << sum << '\n';
system("pause");
}

#include
using namespace std;
int main()
{
int n, c = 0;
cout << "Enter the array size"<< endl;
cin >> n;
float a[n][n];
cout << "Enter the elements of array " << endl;
for (int i = 0; i a[i][j];
}
}
for (int i = 0; i<n; i++)
{
for (int j = 0; j<n; j++)
{
if (i==j)
c = c + a[i][j];
}
}
cout << "Sum = " << c;
}