JZOJ4869【NOIP2016提高A组集训第9场11.7】平均数

Description

Data Constraint

Solution

我们考虑用二分答案解决这个问题。设s[i]表示i的前缀和，那么对于一个二分出来的值x，平均值小于等于x的区间[l+1,r]显然有$(s[r]-s[l])/(r-l)<=x$,移一下项得$s[r]-x*r<=s[l]-x*l$我们记$b[i]=s[i]-x*i$,现在只要求一下b[i]<=b[j]且i>j的数量，即求一下逆序对，用归并排序搞一下就好了，总时间复杂度为O(N$logN^2$)。

Code

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=100005;
double a[maxn],b[maxn],c[maxn],l,r,mid,x;
ll i,j,t,k,n,m;
void dg(int l,int r){
    int mid=(l+r)/2,i=l,j=mid+1,t=l;
    if (l==r) return;
    dg(l,mid);dg(mid+1,r);
    while (i<=mid && j<=r)
        if (a[i]<a[j]) c[t++]=a[i++];
        else c[t++]=a[j++],k+=mid-i+1;
    while(i<=mid) c[t++]=a[i++];
    while(j<=r) c[t++]=a[j++];
    for (i=l;i<=r;i++) a[i]=c[i];
}
int main(){
//  freopen("ave.in","r",stdin);freopen("ave.out","w",stdout);
    scanf("%lld%lld",&n,&m);m=n*(n+1)/2-m+1;
    for (i=1;i<=n;i++)
        scanf("%lf",&x),b[i]=b[i-1]+x;
    l=0;
    r=1e9;
    while (r-l>0.00001){
        mid=(l+r)/2;
        for (i=0;i<=n;i++)
            x=i,a[i]=b[i]-x*mid;
        k=0;
        dg(0,n);
        if (k>=m) r=mid;
        else l=mid+0.00001;
    }
    printf("%.4lf\n",l);
}