CCF NOI1045. 元素之和 (C++)
阿新 • • 發佈:2018-12-11
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;
}