zcmu-4954: 矩阵对角求和（水）

 

Time Limit: 1 Sec  Memory Limit: 32 MB
Submit: 11  Solved: 5
[Submit][Status][Web Board]

Description

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

Input

第一行是一个正整数n。

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

Output

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

Sample Input

3 5724 1478 9358 6962 4464 5705 8145 3281 6827

Sample Output

17015 21967

注意：主对角线是‘\’这个方向 的.

#include <iostream>
#include <cstdio>
using namespace std;
const int maxn = 100 + 5;
int a[maxn][maxn];
int main()
{
    int sum = 0,sum1 = 0;
    int n;
    cin>>n;
    for(int i = 1 ;i <= n ; i++)
    {
        for(int j = 1; j <= n;j++)
            cin>>a[i][j];
    }
    
    for(int i = 1 ;i <= n ; i++)
    {
        for(int j = 1; j <= n;j++)
        {
            if(i == j)
                sum += a[i][j];
            if(n - j + 1 == i)
                sum1 += a[i][j];
        }
    }
    printf("%d\n%d\n",sum,sum1);
    return 0;
}