c++，对角求和！求解答

Question

问题	求一个n×n数字方阵对角线元素之和
输入	第一行：n 第二行至n+1行：数字方阵
输出	对角线元素和

样例输入：

3
1 2 3
1 1 1
3 2 1

样例输出：

9

怎么做？！

Answer 1

供参考：

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