问题 E: Problem B Codeup ContestID：100000581

题目链接：http://codeup.cn/problem.php?cid=100000581&pid=4

题目描述
请写一个程序，对于一个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<stdio.h>
#include<string.h>
#include<algorithm>

using namespace std;

bool cmp(int a, int b) {
	return a > b;
}

int main() {
	int m, a[10][10];
	while(scanf("%d", &m) != EOF){
		int b[30] = {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[i] += a[i][j];
		for(int j = 0; j < m; j++)		//列之和
			for(int i = 0; i < m; i++) 
				b[m + j] += a[i][j];
		for(int i = 0, j = 0; i < m && j < m; i++, j++)		//主对角线之和
			b[2 * m] += a[i][j];
		for(int i = 0, j = m - 1; i < m && j >= 0; i++, j--)	//副对角线之和
			b[2 * m + 1] += a[i][j];
		sort(b, b + 2 * m + 2, cmp);
		for(int i = 0; i < 2 * m + 2; i++)
			printf("%d ", b[i]);
		printf("\n");
	}
	return 0;
}

【注】：OJ通过，为多组数据输入（虽题中未说），否则会答案错误50。