方阵排序

问题 E: Problem B
时间限制: 1 Sec 内存限制: 32 MB
提交: 793 解决: 213
[提交][状态][讨论版][命题人:外部导入]
题目描述
请写一个程序，对于一个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<algorithm>
#include<cstring>
using namespace std;
bool cmp(int a, int b) {
	return a > b;
}
int main() {
	int m, a[10][10], k = 0;
	while (scanf("%d",&m) != EOF) {
		//输入m行m列的方阵
		k = 0;
		int b[22] = { 0 };
		for(int i=0;i<m;i++)
			for (int j = 0; j < m; j++)
			{
				scanf("%d",&a[i][j]);
			}
		//求每一行的元素之和
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < m; j++)
			{
				b[k] += a[i][j];
			}
			k++;
		}
			
		//求每一列的元素之和
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < m; j++)
			{
				b[k] += a[j][i];
			}
			k++;
		}
		
		//求主对角线元素之和
		for (int i = 0; i < m; i++) {
			b[k] += a[i][i];
		}
		//求副对角线元素之和
		for (int i = 0; i < m; i++) {
			b[k + 1] += a[i][m-1 - i];
		}
		sort(b, b + k+2, cmp);
		for (int i = 0; i < k+2; i++)
			printf("%d ", b[i]);
		printf("\n");
	}
	system("pause");
	return 0;
}