Вычислить сумму элементов на главной диагонали матрицы

#include <iostream>
#include <iomanip> 
#include <ctime>
 
using namespace std;
 int main ()
{
    
    srand(time(NULL));
 
    int a[5][5];
    int sum_main_diag = 0;
    int sum_diag = 0;
 
    cout<<"Matrix: "<<endl;
    for(int i=0; i<5; ++i, cout<<endl)
        for(int j=0; j<5; ++j)
        {
            a[i][j]=rand()%21;
            cout<<setw(5)<<a[i][j]<<" ";
        }
 
    cout<<endl;
 
 
    for(int i=0; i<5; ++i)
        sum_main_diag += a[i][i];
 
    cout<<"Sum main diag = "<<sum_main_diag<<endl;
 
 
    for(int i=0, j=4; i<5; ++i, --j)
        sum_diag += a[i][j];
 
    cout<<"Sum diag = "<<sum_diag<<endl;
 
    system("pause");
    return 0;
}

#include <stdio.h>
#include <stdlib.h>
#include <conio.h>
#include <memory.h>
 
#define N 5
 
int main(int argc, char* argv[])
{
    int** A = new int*[N];
    memset((void*)A, 0x00, 4 * N);
 
    for (int t1 = 0; t1 < N; t1++)
    {
        A[t1] = new int[N];
        for (int t2 = 0; t2 < N; t2++)
        {
            A[t1][t2] = rand() % 20;
            printf("%d ",A[t1][t2]);
        }
 
        printf("\n");
    }
 
    int r = N-1, s1 = 0, s2 = s1;
    for (int i = 0; i < N; i++, r--)
    {
        s1+=A[i][i];
        s2+=A[i][r];
    }
 
    printf("\ns1 = %d s2 = %d\n",s1,s2);
 
    _getch();
 
    return 0;
}