6127:Largest Average

#include<bits/stdc++.h>
using namespace std;
int a[100001];
double ave[100001];
struct student{
    int i;
    double ave;
};
student stu[100001];

bool cmp(const student &s1,const student &s2){
    if(s1.ave==s2.ave){
        return s1.i<s2.i;
    }else{
        return s1.ave>s2.ave;
    }
}

int main(){
    priority_queue<int,vector<int>,greater<int> > q;
    int n;
    while(cin>>n){
    memset(stu,0,sizeof(student)*n);
    memset(a,0,sizeof(a));
    memset(ave,0,sizeof(ave));
    for(int i=0;i<n;++i){
        scanf("%d",&a[i]);
    }
    double s=0;
    for(int i=n-1;i>0;i--){
        if(i!=n-1){
            if(a[i]<q.top()){  //如果这个数比队列中最小的值还小,算平均值的时候先没放进队列 
                int ss=q.size();
                ave[i]=s*1.0/ss;
                stu[i].ave=ave[i];
                stu[i].i=i;
                q.push(a[i]);  //最后再放进队列 
            }else{          //如果这个数比队列中最小的值大 
                q.push(a[i]);  //把这个数放进队列 
                int x=q.top();
                q.pop();  //把队列中最小的值弹出 
                int ss=q.size();
                ave[i]=(s+a[i]-x)*1.0/ss;  //算出平均值 
                stu[i].ave=ave[i];
                stu[i].i=i;
                q.push(x);  //最后把最小的值压回去队列 
            }
        }else{
            q.push(a[i]);
            ave[i]=a[i];
            stu[i].ave=ave[i];
            stu[i].i=i;
        }
        s+=a[i];
        //printf("%d %.2f\n",stu[i].i,stu[i].ave);
    }
    sort(stu,stu+n,cmp);
    cout<<stu[0].i;
    for(int i=1;i<n;++i){
        if(stu[i].ave==stu[0].ave&&stu[i].i<=n-2){
            cout<<" "<<stu[i].i;
        }else{
            break;
        }
    }
    while(!q.empty()){
        q.pop();
    } 
    cout<<endl;
    }
}

 

posted @ 2020-09-19 17:17  菜包妄想做肉包  阅读(68)  评论(0)    收藏  举报