洛谷 P1223 排队接水

洛谷 P1223 排队接水

题目描述

有n个人在一个水龙头前排队接水，假如每个人接水的时间为Ti，请编程找出这n个人排队的一种顺序，使得n个人的平均等待时间最小。

输入输出格式

输入格式：

输入文件共两行，第一行为n；第二行分别表示第1个人到第n个人每人的接水时间T1，T2，…，Tn，每个数据之间有1个空格。

输出格式：

输出文件有两行，第一行为一种排队顺序，即1到n的一种排列；第二行为这种排列方案下的平均等待时间(输出结果精确到小数点后两位)。

输入输出样例

输入样例#1：

10 
56 12 1 99 1000 234 33 55 99 812

输出样例#1：

3 2 7 8 1 4 9 6 10 5
291.90

说明

n<=1000

ti<=1e6，不保证ti不重复

思路：

贪心算法，排队接水时从时间小的往后排，等待时间最少，记录下所有人的等待时间的和（注意，第一个人等待时间为0，第二个人的等待时间为第一个人接水的时间，以此类推），求平均值，用结构体输出它的编号。

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

int s[1001],tot,n;
double ans,m;
struct Pd{
    int a,b;
}d[1001];

int cmp(const Pd &p,const Pd &q){
     if(p.b<q.b) return 1;
     else return 0;
}

int main()
{
    cin>>m;
    n=(int)m;
    for(int i=1;i<=n;i++){
        /*d[i].a=d[i-1].a+1;*/
        d[i].a=i;
        cin>>d[i].b;
    }sort(d+1,d+n+1,cmp);
    for(int i=2;i<=n;i++){
        tot+=d[i-1].b;
        s[i]+=tot;
        ans+=s[i];
    }
    /*for(int i=1;i<=n;i++){
        for(int j=1;j<i;j++){
            ans+=d[j].b;
        }
    }*/
    for(int i=1;i<=n;i++){
        cout<<d[i].a<<" ";
    }cout<<endl;
    double w=ans/m;
    printf("%.2lf",w);
    return 0;
}