题目链接:http://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
代码
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
bool cmp(int a, int b) {
return a > b;
}
int main() {
int m, a[10][10];
while(scanf("%d", &m) != EOF){
int b[30] = {0};
for(int i = 0; i < m; i++)
for(int j = 0; j < m; j++)
scanf("%d", &a[i][j]);
for(int i = 0; i < m; i++) //行之和
for(int j = 0; j < m; j++)
b[i] += a[i][j];
for(int j = 0; j < m; j++) //列之和
for(int i = 0; i < m; i++)
b[m + j] += a[i][j];
for(int i = 0, j = 0; i < m && j < m; i++, j++) //主对角线之和
b[2 * m] += a[i][j];
for(int i = 0, j = m - 1; i < m && j >= 0; i++, j--) //副对角线之和
b[2 * m + 1] += a[i][j];
sort(b, b + 2 * m + 2, cmp);
for(int i = 0; i < 2 * m + 2; i++)
printf("%d ", b[i]);
printf("\n");
}
return 0;
}
【注】:OJ通过,为多组数据输入(虽题中未说),否则会答案错误50。