1. 程式人生 > >CCF NOI1045. 元素之和 (C++)

CCF NOI1045. 元素之和 (C++)

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