求一个3×3矩阵对角线元素之和

标题：求一个3×3矩阵对角线元素之和

题目描述
求一个3×3矩阵对角线元素之和。
输入
矩阵
输出
主对角线 副对角线 元素和
样例输入

1 2 3
1 1 1
3 2 1

样例输出

3 7

代码如下：

#include <stdio.h>
#include <stdlib.h>

int main()
{
    int a[3][3],i,j,sum=0,s=0;
    for(i=0;i<3;i++)
    {
        for(j=0;j<3;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    for(i=0;i<3;i++)
    {
        for(j=0;j<3;j++)
        {
            if(i==j)
              sum=sum+a[i][j];
            if(i==0&&j==2)
              s=s+a[i][j];
            if(i==1&&j==1)
              s=s+a[i][j];
            if(i==2&&j==0)
              s=s+a[i][j];
        }
    }
    printf("%d %d",sum,s);
    return 0;
}