C语言题（14）——求矩阵主对角线元素之和

编程实现：求一个5*5整型矩阵主对角线元素之和。

示例

输入

1  2  3  4  5
6  7  8  9  10
11 12 13 14 15
16 17 18 19 20
21 22 23 24 25

输出

主对角线元素之和为65

代码如下：

#include <stdio.h>
#define N 5
int sum(int a[N][N]) {
    int s = 0;
    for (int i = 0; i < N; i++)
        s += a[i][i];
    return s;
}
int main() {
    int a[N][N],s;
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) 
            scanf("%d", &a[i][j]);
    }
    s=sum(a);
    printf("主对角线元素之和为%d\n", s);

    return 0;
}