Description
![这里写图片描述](https://img-blog.csdn.net/20161107201712050)
Data Constraint
![这里写图片描述](https://img-blog.csdn.net/20161107201730112)
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
logN2
)。
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(){
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);
}