Codeforces 939E - Maximize!

939E - Maximize!

思路：

贪心：最后的集合是最大值+前k小个

因为平均值时关于k的凹形函数，所以可以用三分求最小值

又因为后面的k肯定比前面的k大，所以又可以双指针

三分：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mp make_pair
#define pli pair<ll,int>
#define mem(a,b) memset(a,b,sizeof(a))

const int N=5e5+;
ll sum[N],cnt=;
int main(){
    ios::sync_with_stdio(false);
    cin.tie();
    cout.tie();
    cout<<fixed<<setprecision();
    int q,t,x;
    cin>>q;
    while(q--){
        cin>>t;
        if(t==){
            cin>>x;
            sum[++cnt]=sum[cnt-]+x;
        }
        else{
            int l=,r=cnt-,m1=(l+l+r)/,m2=(l+r+r)/;
            while(l<r){
                if((double)(sum[m1]+x)/(m1+)>=(double)(sum[m2]+x)/(m2+)){
                    if(l==m1)break;
                    else l=m1;
                }
                else{
                    if(r==m2)break;
                    else r=m2;
                }
                m1=(l+l+r)/,m2=(l+r+r)/;
            }
            double tt=(double)(sum[l]+x)/(l+);
            for(int i=l;i<=r;i++)tt=min(tt,(double)(sum[i]+x)/(i+));
            cout<<x-tt<<endl;
        }
    }
    return ;
}

双指针:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mp make_pair
#define pli pair<ll,int>
#define mem(a,b) memset(a,b,sizeof(a))

const int N=5e5+;
ll sum[N],cnt=,top=,x;
double cal(int l){
    return (double)(x+sum[l])/(l+);
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie();
    cout.tie();
    cout<<fixed<<setprecision();
    int q,t;
    cin>>q;
    while(q--){
        cin>>t;
        if(t==){
            cin>>x;
            sum[++cnt]=sum[cnt-]+x;
        }
        else{
            while(top<cnt){
                if(cal(top+)<cal(top))top++;
                else break;
            }
            cout<<x-cal(top)<<endl;
        }
    }
    return ;
}