《算法笔记》4.1算法初步 排序 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

总结

计算每行、每列和两条对角线元素之和,存在sum数组中, 再对sum数组排序。注意, 此题是多组输入。

代码

#include<iostream>
#include<algorithm>
using namespace std;

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

int main(int argc, char *argv[]) {
	int matrix[10][10];
	int m;
	while(cin>>m)
	{
	int sum[50] = {0};
	if(m <= 1 || m >= 10)
		break;
	for(int i = 0; i < m; i++)
		for(int j = 0; j < m; j++)
			cin>>matrix[i][j];
	for(int i = 0; i < m; i++)
	{
		for(int j = 0; j < m; j++)
		{
			if(i == j)  //主对角线元素之和 
				sum[2 * m] +=  matrix[i][j];
			if(i == m - j - 1) //副对角线元素之和
				sum[2 * m + 1] +=  matrix[i][j];
			sum[i] += matrix[i][j];	 //每行之和 
			sum[m + i] += matrix[j][i]; //每列之和 
		}
	}
	sort(sum, sum + 2 * m + 2, cmp); 
	for(int i = 0; i < 2 * m + 2; i++)
		cout<<sum[i]<<" ";
	cout<<endl;
	}
	return 0;
}