P1223 排队接水

https://www.luogu.com.cn/problem/P1223

#include<bits/stdc++.h>
using namespace std;
int n;
struct t{  //定义结构体存放等待时间和排队编号 
    int w, no;
};
t a[1005];  //存放n 个人 的数据 
bool cmp(t x, t y){  //定义排序规则：按等待时间从小到大排序 
    return x.w<y.w;
}
long long  sum;  //存放所有人的等待时间 
double ans;//存放答案 
int main()
{
    cin>>n;
    for(int i=1; i<=n; i++)
        cin>>a[i].w, a[i].no=i;  //输入等待时间，记录编号 
    sort(a+1,a+1+n,cmp);         //按等待时间从小到大排序
    for(int i=1; i<=n; i++){
        cout<<a[i].no<<" ";      //输出排队接水顺序 
        sum+=(n-i)*a[i].w;       //后面每个人的排队等待时间 
    }    
    cout<<endl;
    cout<<fixed<<setprecision(2)<<sum*1.0/n;  //输出答案平均等待时间 
    return 0;
 }