用c语言代码，求4*4矩阵的对角线之和

使用指针，函数，数组来实现。

代码如下：

#include <stdio.h>

double fun(double a[][4],int n)
{
    int i,j;
    double *p;
    double sum =0;
    for (i=0;i<n;i++)
    {
        p= a[i];
        for(j=0;j<4;j++,p++)
        {
            if(i==j)
                sum += *p;
        }
    }
    return sum;
}

int main()
{
    double a[4][4];
    int i,j;
    printf("请输入4*4的数组:\n");
    for (i=0;i<4;i++)
    {
        for(j=0;j<4;j++)
            scanf("%lf",&a[i][j]);
    }
    printf("对角线元素和等于%g",fun(a,4));
    return 0;
}