《算法笔记》4.1小节——算法初步-＞排序 问题 E: Problem B

题目描述

请写一个程序，对于一个m行m列的（1<m<10）的方阵，求其每一行，每一列及主对角线元素之和，最后按照从大到小的顺序依次输出。

输入

共一组数据，输入的第一行为一个正整数，表示m，接下来的m行，每行m个整数表示方阵元素。

输出

从大到小排列的一行整数，每个整数后跟一个空格，最后换行。

样例输入 Copy

4
15 8 -2 6
31 24 18 71
-3 -9 27 13
17 21 38 69

样例输出 Copy

159 145 144 135 81 60 44 32 28 27

代码

初出茅庐，请多指教！

#include<stdio.h>
#include<algorithm>
using namespace std;

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

int main(){	
	int m;	
	while(scanf("%d",&m)!=EOF){
		if(m<=1||m>=10){
			printf("m error!\n");
			return 0;
		} 
		int a[10][10];
		int b[35]={0};
		int n=0;
		int i,j;
		for(i=0;i<m;i++)
			for(j=0;j<m;j++){
				scanf("%d",&a[i][j]);
			}
		for(i=0;i<m;i++){
			for(j=0;j<m;j++){
				b[n]=b[n]+a[i][j];
			}
			n++;
		}
		for(j=0;j<m;j++){
			for(i=0;i<m;i++){
				b[n]=b[n]+a[i][j];
			}
			n++;
		}
		for(i=0;i<m;i++){
			b[n]=b[n]+a[i][i];
		}			
		n++;
		
		for(i=0;i<m;i++){		
			b[n]=b[n]+a[i][m-1-i];
		}
		n++;
		sort(b,b+n,cmp);
		for(i=0;i<n;i++){
			printf("%d ",b[i]);
		}
		printf("\n");
	}
	return 0;
}