51nod1711 平均数

Description

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 31 2 3 4 5

Output示例

4.000

思路

首先看题目求区间第k大，首先想到分第k大的平均数x。问题转化为如何快速求任意区间的平均数大于x的个数，先求求前缀和，则可以表示为(sum[r] - sum[l-1])/(r - l +1) > x 的区间个数，转化一下即为sum[r] - r*x > sum[l-1] - (l-1) * x;即形成一个新的数组sum[i]- i * x，求满足条件的区间有多少个，离散化一下，用树状数组统计一下就行。

注意：i从0开始，sum[0] = 0;也要算进去，不然会把0-1的区间漏掉。

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 100010
typedef long long ll;
using namespace std;
ll sum[MAXN];
double a[MAXN],b[MAXN];
int tree[MAXN];
int n;
ll k;

int lowbit( int x)
{
    return x & -x;
}
void add(int k)
{
    while(k <= n + 1)
    {
        tree[k] ++;
        k += lowbit(k);
    }
}
int Sum(int k)
{
    int sum = 0;
    while(k)
    {
        sum += tree[k];
        k -= lowbit(k);
    }
    return sum;
}
long long check(double x)
{
    memset(tree, 0, sizeof(tree));
    for(int i = 0; i <= n; i++)
        a[i] = b[i] = sum[i] - i * x;
    sort(a, a + 1 + n);
    int num = unique(a, a + 1 + n) - a;
    long long ans = 0;
    for(int i = 0; i <= n; i++)
    {
        int temp = lower_bound(a, a + num, b[i]) - a + 1;
        ans += Sum(temp);
        add(temp);
    }
    return ans;
}
int main()
{
    scanf("%d%I64d",&n,&k);
    int temp;
    sum[0] = 0;
    for(int i =  1; i <= n; i++)
    {
        scanf("%d",&temp);
        sum[i] = sum[i - 1] + temp;
    }
    double l = 1.0, r = 100000.0;
    while(r - l >= 1e-6)
    {
        double mid = (l + r) / 2.0;
        if (check(mid) >= k)
            l = mid;
        else
            r = mid;
    }
    printf("%.4f\n",(r + l)/2.0);
    return 0;
}