PAT练习 Problem B

PAT练习 Problem B

Description

请写一个程序，对于一个m行m列的（1<m<10）的方阵，求其每一行，每一列及主对角线元素之和，最后按照从大到小的顺序依次输出。

Input

共一组数据，输入的第一行为一个正整数，表示m，接下来的m行，每行m个整数表示方阵元素。

Output

从大到小排列的一行整数，每个整数后跟一个空格，最后换行。

Sample Input

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

Sample Output

159 145 144 135 81 60 44 32 28 27

C++实现

题目说了只有一组输入数据，答案却需要多组

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
bool cmp(int a,int b){
    return a>b;
}

int main(){
    int m;
    int array[11][11],sum[110];
    while(cin>>m){
    for(int i=0;i<m;i++){
        for(int j=0;j<m;j++){
            cin>>array[i][j];
        }
    }
    for(int i=0;i<2*m+2;i++){
        sum[i]=0;
    }
    for(int i=0;i<m;i++){
        for(int j=0;j<m;j++){
            sum[i]+=array[i][j];
        }
    }
    for(int i=0;i<m;i++){
        for(int j=0;j<m;j++){
            sum[m+i]+=array[j][i];
        }
    }
    for(int i=0;i<m;i++){
        sum[2*m]+=array[i][i];
        sum[2*m+1]+=array[i][m-i-1];
    }
    sort(sum,sum+2*m+2,cmp);
    for(int i=0;i<2*m+2;i++){
        cout<<sum[i]<<" ";
        
    }
    cout<<endl;
    }
}