问题 E: Problem B
时间限制 : 1.000 sec 内存限制 : 32 MB
题目描述
请写一个程序,对于一个m行m列的(1<m<10)的方阵,求其每一行,每一列及主对角线元素之和,最后按照从大到小的顺序依次输出。
输入
共一组数据,输入的第一行为一个正整数,表示m,接下来的m行,每行m个整数表示方阵元素。
输出
从大到小排列的一行整数,每个整数后跟一个空格,最后换行。
样例输入
4
15 8 -2 6
31 24 18 71
-3 -9 27 13
17 21 38 69
样例输出
159 145 144 135 81 60 44 32 28 27
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int a[11][11];
int main() {
int n;
while(cin >> n){
vector<int> sum;
int sum_row = 0, sum_line = 0, sum_diagonal = 0, sum_minus_diagonal = 0;
for (int i = 0; i < n; ++i) {
sum_row = 0;
for (int j = 0; j < n; ++j) {
cin >> a[i][j];
sum_row += a[i][j];
}
sum.push_back(sum_row);
}
for (int i = 0; i < n; ++i) {
sum_line = 0;
for (int j = 0; j < n; ++j) {
sum_line += a[j][i];
}
sum.push_back(sum_line);
}
for (int i = 0; i < n; ++i) {
sum_diagonal += a[i][i];
}
for (int i = 0, j = n-1; i < n; ++i, j--) {
sum_minus_diagonal += a[i][j];
}
sum.push_back(sum_diagonal);
sum.push_back(sum_minus_diagonal);
sort(sum.begin(), sum.end(), greater<int>());
for (auto s:sum) {
cout << s << " ";
}
cout << endl;
}
return 0;
}