二维数组:求方阵主、副对角线上元素之和

二维数组:求方阵主、副对角线上元素之和
二维数组:求5行5列矩阵的主、副对角线上元素之和。 注意:两条对角线相交的元素只加一次。

函数接口定义:
int fun(int a[M][M]) ;
函数返回主副对角线元素的累加和。

裁判测试程序样例:

#include <stdio.h>
#define M 5
int fun(int a[M][M]) ;

int main()
{
int a[M][M],i,j;
int y;
for(i=0;i<M;i++)
for(j=0; j<M; j++)
scanf("%d", &a[i][j]) ;
y=fun(a);
printf("sum=%d", y) ;
return 0;
}

/* 请在这里填写答案 */
输入样例:
1 3 5 7 19
2 4 6 1 10
1 2 3 4 15
2 5 6 2 18
3 3 8 8 12
输出样例:
sum=50

代码如下:


int fun(int a[M][M])
{
    int i=0,j=0;
    int sum = 0;
    for (i=0;i<M;i++)
    {
        for (j=0;j<M;j++)
        {
            if(i==j)
                sum += a[i][j];  //主对角线
            else if (i+j ==M-1)
            {
                sum += a[i][j];  //副对角线
            }
        }
    }
    return sum;
}