求矩阵的对角线之和

【问题描述】编程计算n*n矩阵的主对角线元素和反向对角线元素之和。（要求①编写函数实现，且将主对角线元素和反向对角线元素之和作为函数值返回；②n*n矩阵各元素要求通过键盘输入）

【输入形式】n的值  n*n个整数
【输出形式】输出该n*n矩阵的主对角线元素和反向对角线元素之和

【样例输入】3

                    4 7 9 1 8 2 3 2 6
【样例输出】sum=30

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