2016年26．从键盘上输入矩阵的阶数n (n＜5)，再输入矩阵的各个数据元素，计算矩阵的对角线元素之和 sum。按照下面格式输出该矩阵和sum的值。

/*
	26．从键盘上输入矩阵的阶数n (n<5)，再输入矩阵的各个数据元素，计算矩阵的对角线元素之和 sum。按照下面格式输出该矩阵和sum的值。
	例如n=3时，矩阵为;
	1 3 6
	3 6 9
	6 9 12
	sum = 19
*/

#include <stdio.h>
#include <string.h>

int main() {
	int n, i, j, sum = 0;
	int arr[10][10];
	scanf("%d", &n);
	for (i = 0; i < n; ++i) {
		for (j = 0; j < n; ++j) {
			scanf("%d", &arr[i][j]);
			if (i == j) {
				sum += arr[i][j];
			}
		}
	}
	printf("sum = %d\n", sum);
	return 0;
}