矩阵对角线元素之和

输入一个NxN的方阵,求对角线上元素的和. (说明,N>0,N<=10,矩阵元素为整型数),若输入数据不满足要求,则输出error. input: 输入N以及NxN个元素 output: 输出对角线上元素之和 example: input: 3 1 2 3 4 5 6 7 8 9 output: 15

输入格式:

3 1 2 3 4 5 6 7 8 9

输出格式:

15

输入样例:

3
1 2 3
4 5 6
7 8 9

结尾无空行

输出样例:

15

结尾无空行

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