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;
}