Codeup墓地 Contest100000581 问题 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<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
bool cmp(int a,int b){
	return a>b;
}
int main(){
	int m;
	int array[10][10];
	int ans[22];
	while(scanf("%d",&m)!=EOF){
		int cnt=0;
		memset(ans,0,sizeof(ans));
		for(int i=0;i<m;i++){//录入方阵并计算每一行的和
			for(int j=0;j<m;j++){
				scanf("%d",&array[i][j]);
				ans[cnt]+=array[i][j];
			}
			cnt++;
		}
		for(int i=0;i<m;i++){//计算每一列的和
			for(int j=0;j<m;j++){
				ans[cnt]+=array[j][i];
			}
			cnt++;
		}
		for(int i=0;i<m;i++){//计算对角线的和
			ans[cnt]+=array[i][i];
			ans[cnt+1]+=array[m-i-1][i];
		}
		sort(ans,ans+2*m+2,cmp);//排序
		for(int i=0;i<2*m+2;i++){//输出结果
			printf("%d ",ans[i]);
		}
		printf("\n");
	}
	return 0;
}