《算法筆記》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;
}