OJ数组求对角线之和

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

Input
矩阵

Output
主对角线 副对角线 元素和

Sample Input
1 2 3
1 1 1
3 2 1
Sample Output
3 7
#include <iostream>

using namespace std;

int main()
{

    int i,j,a[3][3],s1=0,s2=0;

    for (i=0;i<3;i++)

        for (j=0;j<3;j++)

            cin>>a[i][j];
        for(i=0;i<3;i++)
        {
            s1+=a[i][i];
            s2+=a[i][2-i];
        }

    cout<<s1<<" "<<s2;

    return 0;
}