hdu-1031-Design T-Shirt

原创于 2014-07-25 11:35:13 发布 · 401 阅读

CC 4.0 BY-SA版权

本文介绍了一个使用C++实现的程序，该程序通过数组和结构体来处理和排序数据。具体而言，程序首先读取输入数据到二维数组中，并计算每列的平均值，然后将这些平均值及对应的原始索引保存到结构体数组中进行两次排序：第一次按平均值降序排列，第二次对前k个元素按原始索引升序排列。

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
double a[500][500];
struct in
{
    double x;
    int y;
}b[500];
int cmp1(const in &a,const in &b)
{
    //struct in *c=(in *)a;
   // struct in *d=(in *)b;
    return a.x>=b.x;//c->x<d->x;
}
int cmp2(const in &a,const in &b)
{
    //struct in *c=(in *)a;
    //struct in *d=(in *)b;
    return b.y<a.y;//d->y-c->y;
}
int main()
{
    int n,m,k,i,j;
    double sum;
    while(scanf("%d%d%d",&n,&m,&k)!=EOF)
    {
        for(i=0;i<n;i++)
         for(j=0;j<m;j++)
         scanf("%lf",&a[i][j]);
         for(j=0;j<m;j++)
         {
            for(sum=0.0,i=0;i<n;i++)
            sum+=a[i][j];
            b[j].x=sum/n;
            b[j].y=j+1;
        }
        sort(b,b+m,cmp1);
        sort(b,b+k,cmp2);
        for(i=0;i<k-1;i++)
        printf("%d ",b[i].y);
        cout<<b[k-1].y<<endl;
    }
    return 0;
}