矩阵相乘——矩阵相乘的轨迹（C语言）

问题描述
设有NxN阶矩阵A，那么矩阵A的迹（用tr(A)表示）就等于A的特征值的总和，也即矩阵A的主对角线元素（左上到右下方向）的总和。
例如2阶矩阵A为
2 3
4 5
其迹为tr(A)=2+5=7
现在要求求取两个N阶矩阵A与B乘积的迹，也即tr(AB).
输入说明
有2N+1行输入：
第1行为矩阵的阶数N（N的值 介于1到10之间）
第2~N+1行为矩阵A，矩阵中每个数值均为整数，以空格分开；
第N+2~2N+1行为矩阵B，矩阵中每个数值均为整数，以空格分开。
输出说明
一个值，即tr(AB)。
输入样例
2
1 1
1 1
1 0
0 1
输出样例
2

矩阵相乘：

#include<stdio.h>
int main()
{
	int i,j,k,n,sum,sum2=0,a[10][10],b[10][10],c[10][10];
	scanf("%d",&n);
	for(i=0;i<n;i++)
	{
		for(j=0;j<n;j++)
			scanf("%d",&a[i][j]);
	}
	for(i=0;i<n;i++)
	{
		for(j=0;j<n;j++)
			scanf("%d",&b[i][j]);
	}
	for(i=0;i<n;i++)                   //利用三重的循环进行矩阵相乘
	{
		for(j=0;j<n;j++)
		{
			sum=0;
			for(k=0;k<n;k++)
				sum += a[i][k]*b[k][j];
			c[i][j]=sum;
		}
	}
	for(i=0;i<n;i++)
		sum2 += c[i][i];
	printf("%d",sum2);
	return 0;
 } 

欢迎交流探讨。
谢谢。