【结构体排序】HDU-1031 Design T-Shirt

注解

1、结构体排序
2、vector使用

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>

using namespace std;

struct tshirt {
    int num;
    double sum;
};

int compare(tshirt a, tshirt b) {
    if(a.sum!=b.sum) {
        return a.sum>b.sum;
    } else {
        return a.num<b.num;
    }

}

int main() {

    int N, M, K;
    while(cin>>N>>M>>K) {
        tshirt a[M];
        memset(a, 0, sizeof(a));
        for(int j=0; j<N; j++) {
            for(int i=0; i<M; i++) {
                a[i].num = i+1;
                double tmp;
                cin>>tmp;
                a[i].sum += tmp;
            }
        }
        sort(a, a+M, compare);
        vector<int> v;
        for(int i=0; i<K; i++) {
            v.push_back(a[i].num);
        }
        sort(v.begin(), v.end());
        cout<<v.at(v.size()-1);
        for(int i=v.size()-2; i>=0; i--){
            cout<<" "<<v.at(i);
        }
        cout<<endl;
    }

    return 0;
}

结果