1045. 元素之和

題目描述

輸入4*4方陣，分別求兩條對角線上元素之和。

輸入

輸入4*4方陣。

輸出

輸出兩條對角線上元素之和(如樣例所示)。

樣例輸入

0 0 2 7
5 3 2 1
9 9 7 0
9 1 9 5

樣例輸出

15 27

資料範圍限制

0<元素值<10000

C++程式碼

#include <iostream>
#include <cassert>

using namespace std;

const int max_size = 4;

int main()
{
    int
 
 matrix[max_size][max_size];

    for(int row=1; row<=max_size; row++)
    {
        for(int col=1; col<=max_size; col++)
        {
            cin >> matrix[row-1][col-1];
            assert (matrix[row-1][col-1] >= 0 && matrix[row-1][col-1] <= 10000);
        }
    }

    int sum_tl2br =
 
 0; // elements' sum from top left down to bottom right 
    int sum_bl2tr = 0; // elements' sum from bottom left up to top right

    for(int row=1; row<=max_size; row++)
    {
        for(int col=1; col<=max_size; col++)
        {
            if (row == col)
            {
                sum_tl2br +
 
= matrix[row-1][col-1];
            }
            else if (row + col == max_size+1)
            {
                sum_bl2tr += matrix[row-1][col-1];
            }
        }
    }

    cout << sum_tl2br << " " << sum_bl2tr << endl;

    return 0;
}