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

版权声明：copyright©CodeIover reserved https://blog.csdn.net/qq_40073459/article/details/86599211

                                          问题 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<algorithm>
#include<cstring>
using namespace std;
bool cmp(int a,int b)
{
	return a>b;
}
int main()
{	
    int m;
   while(scanf("%d",&m)!=EOF)
   {
    int a[m][m];
    for(int i=0;i<m;i++)
      for(int j=0;j<m;j++)
        scanf("%d",&a[i][j]);
    int row[m]={0},column[m]={0},dui=0,fudui=0;
    for(int i=0;i<m;i++)
    {
	  dui+=a[i][i];//主对角线元素和 
	  fudui+=a[i][m-1-i];//副对角线元素和 
      for(int j=0;j<m;j++)
         {
		   row[i]+=a[i][j];//各行元素和 
           column[i]+=a[j][i];//各列元素和 
          }
    }
    int sum[2*m+2]={0};
    int i,j=0,k=0;
    for(i=0;i<2*m;i++)
    {
    	if(i<m) {
    		sum[i]=row[j++];
		}
		if(m<=i&&i<2*m) 
		{
			sum[i]=column[k++];
		}
	}
	sum[2*m]=dui;
	sum[2*m+1]=fudui;
	sort(sum,sum+2*m+2,cmp);
	for(int i=0;i<2*m+2;i++)
	{
		printf("%d ",sum[i]);
	}
	printf("\n");
  }
    return 0;
	
}

 

结果如下：