算法笔记 — Problem B

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/m0_37345402/article/details/84311260

题目链接:http://www.codeup.cn/problem.php?cid=100000581&pid=4

题目描述

请写一个程序,对于一个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

记录每行、每列、主对角线上的数字之和,并将它们放在一个数组中,sort排序即可,也可以逆序输出。

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
bool cmp(int a,int b){
    return a>b;
}
int main(){
    int m; 
	int a[11][11];
	int row[11],col[11],dia[3];
	int ans[111]; 
    while(cin>>m){
    	memset(row,0,sizeof(row));
		memset(col,0,sizeof(col));
		memset(dia,0,sizeof(dia));
		memset(ans,0,sizeof(ans));
		memset(a,0,sizeof(a));
    	for(int i=0;i<m;i++){
	        for(int j=0;j<m;j++){
	            cin>>a[i][j];
	        }
		         
	    }
	    int cnt=0;
	    for(int i=0;i<m;i++){
	        for(int j=0;j<m;j++){
	            row[i]+=a[i][j];
	            col[j]+=a[i][j]; //col[i]+=a[j][i]
	            if(i==j){
	                dia[1]+=a[i][j];
	            }
	            if(i+j==m-1){
	                dia[2]+=a[i][j];
	            }
	        }
	    }
	    for(int i=0;i<m;i++){
	        ans[cnt++]=row[i];
	        ans[cnt++]=col[i];
	    }
	    ans[cnt++]=dia[1];
	    ans[cnt]=dia[2];
	    sort(ans,ans+cnt+1,cmp);
	    for(int i=0;i<=cnt;i++){
	        cout<<ans[i]<<" ";
	    }
	    cout<<endl;
	}
    return 0;
}

猜你喜欢

转载自blog.csdn.net/m0_37345402/article/details/84311260