Problema de codificação B

Descrição do título

Por favor, escreva um programa para uma matriz quadrada com m linhas em colunas (1 <m <10), encontre a soma de cada linha, cada coluna e os principais elementos diagonais e, finalmente, produza na ordem do maior para o menor.

entrar

Há um conjunto de dados. A primeira linha da entrada é um número inteiro positivo, representando m, e as próximas m linhas, m inteiros em cada linha representam elementos da matriz quadrada.

Resultado

Uma linha de inteiros organizados do maior para o menor, cada inteiro seguido por um espaço e, finalmente, uma nova linha.

Cópia de entrada de amostra

4 
15 8 -2 6 
31 24 18 71 
-3 -9 27 13 
17 21 38 69

Exemplo de cópia de saída

159 145 144 135 81 60 44 32 28 27

 

Código AC

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

//res为结果数组
int a[10][10];
int res1[10];
int res2[10];
int res[10];

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

int main()
{
    int m;
    while (scanf("%d",&m) != EOF)
    {
        int j;
        //接收元素
        for (int i = 0;i < m;i++)
            for (int j = 0;j < m;j++)
                scanf("%d",&a[i][j]);
                
        //每一行元素和,第i行元素和存放在res[i]中
        //注意:对于多组测试数据,每次要初始化数组,下面同理
        memset(res1,0,sizeof res1);
        for (int i = 0;i < m;i++)
            for (int j = 0;j < m;j++)
                res1[i] += a[i][j];
                
        //每一列元素和
        memset(res2,0,sizeof res2);
        for (int i = 0;i < m;i++)
            for (int j = 0;j < m;j++)
                res2[i] += a[j][i];
                
        //主对角线元素和
        int sum1 = 0;
        for (int i = 0;i < m;i++)
        {
            j = i;
            sum1 += a[i][j];
        }
        
        //副对角线元素和
        int sum2 = 0;
        for (int i = 0;i < m;i++)
        {
            j = m - i - 1;
            sum2 += a[i][j];
        }
        
        //将所有数据搬到res结果数组中
        j = 0;
        for (int i = 0;i < m;i++) res[j++] = res1[i];
        for (int i = 0;i < m;i++) res[j++] = res2[i];
        res[j++] = sum1;
        res[j++] = sum2;
        sort(res,res + j,cmp);
        for (int i = 0;i < j;i++) printf("%d ",res[i]);
        puts("");
    }
    return 0;
}