编写一个求二维数组主对角线元素和的函数sum( )，在主函数中初始化数组 调用sum函数并在主函数中输出结果

Question

编写一个求二维数组主对角线元素和的函数sum( )，在主函数中定义一个4×4的二维数组，并为其初始化以下数据，
调用sum函数并将结果返回到主函数中输出。
11 22 33 44
25 35 45 55
18 28 38 48
55 66 77 88
c++

Answer 1

#include <iostream>
using namespace std;
int sum(int a[][4])
{
int i,j,s=0;
for(i=0;i<4;i++)
for(j=0;j<4;j++)
if(i==j) s+=a[i][j];
return s;
}
int main()
{
int a[][4]={11,22,33,44,25,35,45,55,18,28,38,48,55,66,77,88};
cout<<sum(a);
return 0;
}

Answer 2

#include <iostream>

const int N = 4;
int sumOfDiagonal(int num[][N])
{
    int res = 0;
    for(int i = 0; i < N; ++i)
        res += num[i][i];
    
    return res;
}

int main()
{
    int a[N][N] = {11, 22 ,33 ,44,
                   25, 35 ,45 ,55,
                   18, 28, 38, 48,
                   55, 66, 77, 88};
    int res = sumOfDiagonal(a);    
    
    std::cout << res << std::endl;
    return 0;
}