问题 E: Problem B

题目描述

请写一个程序，对于一个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<bits/stdc++.h> using namespace std; bool cmp(int a ,int b) { return a > b; } int main() { int m; while(cin >> m) { vector<vector<int>>matrix(m, vector<int>(m)); for(int i=0;i<m;i++) { for(int j=0;j<m;j++) { cin >> matrix[i][j]; } } vector<int>ans(2*m+2, 0); int index = 0; for(int i=0;i<m;i++) { for(int j=0;j<m;j++) { ans[index] += matrix[i][j]; } index++; } for(int i=0;i<m;i++) { for(int j=0;j<m;j++) { ans[index] += matrix[j][i]; } index++; } for(int i=0,j=0;i<m;) { ans[index] += matrix[i][j]; i++; j++; } index++; for(int i=0, j=m-1;i<m;) { ans[index] += matrix[i][j]; i++; j--; } sort(ans.begin(), ans.end(), cmp); for(int i=0;i<2*m+2; i++) { cout<<ans[i]; if(i==2*m+1) cout<<endl; else cout<<" "; } } return 0; }