生成一个n*n的方阵，然后输出此方阵对角线上元素之和

//输入一个正整数n(1<n<10),根据以下公式生成一个n*n的方阵，然后输出此方阵对角线上元素之和

//公式为a[i][j]=i*n+j+1 (0=<i<n,0=<j<n)

源代码：

#include<stdio.h>
#include<stdlib.h>
#define N 10

//输入一个正整数n(1<n<10),根据以下公式生成一个n*n的方阵，然后输出此方阵对角线上元素之和
//公式为a[i][j]=i*n+j+1 (0=<i<n,0=<j<n)

int main()
{
	int a[N][N]={0};
	int sum1 = 0;//主对角线 
	int sum2 = 0;//副对角线 
	for(int i=0;i<N;i++)
	{
		for(int j=0;j<N;j++)
		{
			a[i][j]=i*N+j+1;
			printf("%d	",a[i][j]);//一边赋值一边打印输出 
		}
		printf("
");
	}
	//求对角线元素之和
	for(int i=0;i<N;i++)
	{
		sum1 += a[i][i];//主对角线元素累加 
		if(i!=(N-1-i))//为了防止最中间那个元素已被累加 
		{
			sum2 +=a[i][N-1-i];//副对角线元素累加 
		}

	} 
	printf("两条对角线元素之和sum=%d
",(sum1+sum2));
	
	return 0;
} 

如何分别求主对角线和副对角线则if(i!=(N-1-i))那个判断就不需要了。