数组: 矩阵对角元素之和

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

输入要求：矩阵‪‬‪‬‪‬‪‬‪‬‮‬‫‬‭‬‪‬‪‬‪‬‪‬‪‬‮‬‭‬‫‬‪‬‪‬‪‬‪‬‪‬‮‬‭‬‫‬‪‬‪‬‪‬‪‬‪‬‮‬‭‬‫‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‫‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‪‬

输出要求：分别输出主对角线和副对角线的元素和

#include <iostream>
#include <stdlib.h>
#include <time.h>
using namespace std;
int main()
{
    int a[3][3];
    int i, j, n = 0, m = 0;
    for (i = 0;i < 3;i++)
    {
        for (j = 0;j < 3;j++)
        {
            cin >> a[i][j];
        }
    }
    for (i = 0;i < 3;i++)
    {
        for (j = 0;j < 3;j++)
        {
            if (i == j)
                n += a[i][j];
            if (i + j == 2)
                m += a[i][j];
        }
    }
    cout << n << " " << m << endl;
    return 0;
}