Viết chương trình nhập ma trận vuông các số nguyên AA có nn dòng và nn cột từ bàn phím. Tính và in ra tổng giá trị của các phần tử trên đường chéo chính của ma trận. Biết rằng đường chéo chính là đường xuất phát từ chỉ số (0, 0) và kết thúc ở chỉ số (n-1, n-1) Dữ liệu vào Dòng đầu tiên là số nguyên dương nn Dòng thứ 2 là n * nnn số nguyên tương ứng là các phần tử của mảng AA Giới hạn 2n100 |Aij| 10000 Dữ liệu ra Kết quả của bài toán

#include <iostream>
using namespace std;

int main()
{
    int n;
    cin >> n;
    int a[n][n];
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            cin >> a[i][j];
    int sum = 0;
    for (int i = 0; i < n; i++)
        sum += a[i][i];
    cout << sum << endl;
    return 0;
}

#include <bits/stdc++.h>

using namespace std;

long long a[1005][1005],n,s=0;
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++) cin>>a[i][j];
    }
    for(int i=1;i<=n;i++) s+=a[i][i];
    cout<<s;
}