HDU-1031-Design T-Shirt

#include<iostream>
#include<string>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
/*
    题目的意思是告诉你n,m,l,分别表示多少个人参与，多上种元素，最后选多少种元素；
    选取元素的条件就是每个元素最后的得分得尽可能高；
    所以很显然题目就是让你求每个元素最后的总分，并对其排序；
    题目要求输出编号不能递增，并且如果分数一致，尽可能的输出编号小的；
    这个题目我们就用到了两次排序；
*/
//  用结构体定义是为了让下标不变，更容易输出；
struct node
{
    double sum=0.0;
    int id;
};
//  自定义排序1，各元素由得分高到得分低输出，得分相同的优先输出编号小的；
bool cmp(node x,node y)
{
    if(x.sum!=y.sum) return x.sum>y.sum;
    else return x.id<y.id;
}
//  自定义排序2，将最后结果逆序，有编号大的先输出；
bool cmp1(node x,node y)
{
    return x.id>y.id;
}
double a[1005][1005];   //  好吧，这道题没有说名数据量，大概的跑了一个，够用就好了；
int main()
{
    int n,m,l;
    while(~scanf("%d%d%d",&n,&m,&l)){
        node b[1005];
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++) scanf("%lf",&a[i][j]); //  二维数组存放数据；
        }
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++) b[i].sum+=a[j][i]; //结构体编辑计算各元素总得分；
            b[i].id=i;  //  标记编号；
        }
        sort(b,b+m,cmp);    //  自定义排序1；
        sort(b,b+l,cmp1);   //  自定义排序1；
        for(int i=0;i<l-1;i++) printf("%d ",b[i].id+1);
        printf("%d\n",b[l-1].id+1);
    }
    return 0;
}