c 求一个n×n的整形矩阵从左上角至右下角的对角线元素之和

#include <stdio.h>
int main() {
    int n, i, j, sum = 0; scanf("%d", &n);
    int a[n][n];
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            scanf("%d",&a[i][j]);
            if (i==j) sum+=a[i][j];
        }
    }
    printf("%d\n", sum);
    return 0;
}