输入一个二阶矩阵并计算主对角线元素的和 C语言

题目：从键盘上输入矩阵的阶数n（n<5），再输入矩阵的各个数据元素，计算矩阵的对角线元素之和sum。

    按照下面格式输出该矩阵和sum的值。

    例如n=3时，矩阵为：

    1 3 6

    3 6 9

    6 9 12

    sum =19

#define X 5

int main(void) {
	int a[X][X] = { 0 };
	int n = 0;
	int sum = 0;
	printf("请输入矩阵阶数：");
	scanf("%d", &n);
	if (n < 1 || n > X) {
		printf("error input!\n");
		return 0;
	}
	for (int i = 0; i < n; i++) {
		printf("=====第%d行=====\n",i+1);
		for (int j = 0; j < n; j++) {
			scanf("%d", &(a[i][j]));
			if(i==j) sum += a[i][j];
		}
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			printf("%d ", a[i][j]);
			if (j == n - 1) {
				printf("\n");
			}
		}
	}
	printf("sum = %d\n",sum);
	return 0;
}