AC日记——数据流中的算法 51nod 1785

数据流中的算法

思路：

　　线段树模拟；

　　时间刚刚卡在边界上，有时超时一个点，有时能过；

来，上代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define maxn 1000005

int n,size,cnt,ai[maxn],L[maxn<<],R[maxn<<];
int mid[maxn<<],dis[maxn<<],to,x;

double sum[maxn<<],sum2[maxn<<];

inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'') Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}

void tree_build(int now,int l,int r)
{
    L[now]=l,R[now]=r;
    if(l==r) return ;
    mid[now]=l+r>>;
    tree_build(now<<,l,mid[now]);
    tree_build(now<<|,mid[now]+,r);
}

void tree_add()
{
    int now=,dis1=to*x,ss=to*dis1;
    while(L[now]!=R[now])
    {
        dis[now]+=x,sum[now]+=dis1,sum2[now]+=ss;
        if(to<=mid[now]) now=now<<;
        else now=now<<|;
    }
    dis[now]+=x,sum[now]+=dis1,sum2[now]+=ss;
}

int tree_query(int k)
{
    int now=;
    while(L[now]!=R[now])
    {
        if(k<=dis[now<<]) now=now<<;
        else k-=dis[now<<],now=now<<|;
    }
    return L[now];
}

inline void out(int X)
{
    if(X>) out(X/);
    putchar(X%+);
}

int main()
{
    int now=;in(n),in(size);
    int ty;tree_build(,,);
    while(n--)
    {
        in(ty);
        if(ty==)
        {
            in(ai[++now]);
            to=ai[now],x=,tree_add();
            if(now>size) x=-,to=ai[now-size],tree_add();
            continue;
        }
        if(ty==)
        {
            double res=sum[]/(double)dis[];
            out((int)res),putchar('.'),putchar(''),putchar(''),putchar('\n');
            continue;
        }
        if(ty==)
        {
            double p=sum[]/(double)dis[];
            double res=(sum2[]-sum[]**p+p*p*dis[])/dis[];
            printf("%.2lf",res),putchar('\n');
            continue;
        }
        if(ty==)
        {
            int p=dis[];
            if(p&) out(tree_query((p>>)+)),putchar('.'),putchar(''),putchar(''),putchar('\n');
            else printf("%.2f",(double)(tree_query(p>>)+tree_query((p>>)+))/2.0),putchar('\n');
        }
    }
    return ;
}