51Nod-1711平均数(二分+树状数组|线段树)

来源:互联网 发布:ubuntu 安装lnmp 编辑:程序博客网 时间:2024/05/29 12:22

1711 平均数
基准时间限制:4 秒 空间限制:131072 KB 分值: 80 难度:5级算法题
LYK有一个长度为n的序列a。
他最近在研究平均数。
他甚至想知道所有区间的平均数,但是区间数目实在太多了。
为了方便起见,你只要告诉他所有区间(n*(n+1)/2个区间)中第k大的平均数就行了。
Input
第一行两个数n,k(1<=n<=100000,1<=k<=n*(n+1)/2)。
接下来一行n个数表示LYK的区间(1<=ai<=100000)。
Output
一行表示第k大的平均数,误差不超过1e-4就算正确。
Input示例
5 3
1 2 3 4 5
Output示例
4.000
alpq654321 (题目提供者)

题意:

先把k开成longlong。。。算了一下m的数量级是1e9开了intWA了无数次。。。

很容易看出来是二分答案,那么对于每一个答案,有多少个区间满足怎么处理?
(sum[R]sum[L1])/(RL+1)>=ans=>sum[R]ansR>=sum[L1]ans(L1)
形式完全一样,那么以sum[X]-ans*X扔到树状数组,求有多少个数小于等于他就可以了
需要注意的是要把0也要扔进去即sum[0]-ans*0,不然区间统计会错误

#include<cstdio>#include<cstring>#include<algorithm>#include<iostream>#include<cmath>using namespace std;typedef long long LL;const int maxn=1e5+10;const double eps=1e-6;int T[maxn];int kase,n,a[maxn];LL m,c[maxn];double b[maxn],h[maxn];inline LL get(int x){  LL res=0;  for(;x;x-=x&-x)res+=T[x];  return res;}inline void add(int x,int val){  for(;x<=n+1;x+=x&-x)T[x]+=val;}inline LL check(double res){  memset(T,0,sizeof(T));  for(int i=0;i<=n;++i)b[i]=h[i]=c[i]-i*res;  sort(h,h+n+1);  int d=unique(h,h+n+1)-h;  LL cnt=0;  for(int i=0;i<=n;++i)  {    int x=lower_bound(h,h+d,b[i])-h+1;    cnt+=get(x);    add(x,1);  }  return cnt;}inline double find(LL k){  double L=1,R=100000;  while(R-L>=eps)  {    double mid=(L+R)/2.0;    if(check(mid)<k)R=mid;    else L=mid;  }  return (L+R)/2.0;}int main(){  while(~scanf("%d%lld",&n,&m))  {    c[0]=0;    for(int i=1;i<=n;++i)    {      scanf("%d",a+i);      c[i]=c[i-1]+(double)a[i];     }    double ans=find(m);    printf("%.3f\n",ans);  }  return 0;}