AK acm/oj.求矩阵两对角线上的元素之和

题目描述

求矩阵的两对角线上的元素之和

输入
矩阵的行数N
和一个N∗N的整数矩阵ai,j1≤N≤10

输出
所输矩阵的两对角线上的元素之和

样例输入

3
1 2 3 
4 5 6 
7 8 9

样例输出

25

示例代码

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n, sum = 0;
    cin >> n;
    int a[n][n] = {0, 0};
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            cin >> a[i][j];
            if (i == j)
                sum += a[i][j];
            if (i + j == n - 1)
                sum += a[i][j];
        }
    }
    if (n % 2 == 1)
    sum-=a[n/2][n/2];
    cout << sum;
    return 0;
}