问题 1757: 矩阵对角求和

题目链接：https://www.dotcpp.com/oj/problem1757.html

题目描述

求一个n×n矩阵对角线元素之和，其中1≤n<100，矩阵元素都小于10000。

输入

第一行是一个正整数n。

然后是n行，每行对应矩阵的每行，元素间用空格隔开。

输出

第一行为主对角线元素和，第二行为副对角线的元素和。

样例输入

3
5724 1478 9358
6962 4464 5705
8145 3281 6827

样例输出

17015
21967

#include <iostream>
#include <algorithm>
#include <cmath> 
#include <string>
#include <cstring>
#include <map> 
#include <cstdio>
using namespace std;
int a[105][105]; 
int n; 
int main()
{
    while(cin>>n){
        int sum1=0,sum2=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                cin>>a[i][j];
            } 
        } 
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(i==j){
                    sum1+=a[i][j];
                }
                if(i+j==n-1){
                    sum2+=a[i][j];
                }
            }
        }
        cout<<sum1<<endl;
        cout<<sum2<<endl;
    }
    return 0;
}