基础题——求矩阵的主对角线和副对角线和

基础题——求矩阵的主对角线和副对角线和
题目：
求一个3×3矩阵对角线元素之和。
输入：
矩阵
输出：
主对角线 副对角线 元素和
样例输入：
1 2 3
1 1 1
3 2 1
样例输出：
3 7
题目来源：c语言网
代码实现：

#include"stdio.h"
#define num 3
int main (){
	int a[num][num],total1=0,total2=0;
	for(int i=0;i<num;i++)
		for(int j=0;j<num;j++)
			scanf("%d",&a[i][j]);
	for(int i=0;i<num;i++)
		for(int j=0;j<num;j++){
			if(i==j) total1+=a[i][j];
			if(i+j==2)total2+=a[i][j];
		}
	printf("%d %d",total1,total2);
	return 0;
	
} 

使用宏定义的好处：当数组大小改变时，只需要更改宏定义上的数组，防止有地方遗漏。