Codeforces 923 B. Producing Snow

http://codeforces.com/contest/923/problem/B

题意：

有n天，每天产生一堆体积为Vi的雪，每天所有雪堆体积减少Ti

当某一堆剩余体积vi<=Ti时，体积减少vi，雪堆消失

问每天所有雪堆一共减少多少体积

fhq treap

把<=Ti的分裂出来，计算和

>Ti 的 Ti*个数

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

using namespace std;

#define N 100005

int a[N],t[N];
int id[N];

int tot;
int root,fa[N],ch[N][],pri[N];
int siz[N],val[N];
long long sum[N],tag[N];

void read(int &x)
{
    x=; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*+c-''; c=getchar(); }
}

int newnode(int v)
{
    siz[++tot]=;
    sum[tot]=v;
    val[tot]=v;
    pri[tot]=id[tot];
    return tot;
}

void update(int x)
{
    siz[x]=siz[ch[x][]]+siz[ch[x][]]+;
    sum[x]=sum[ch[x][]]+sum[ch[x][]]+val[x];
}

void tagging(int x,int y)
{
    val[x]-=y;
    sum[x]-=1LL*siz[x]*y;
    tag[x]+=y;
}

void down(int x)
{
    if(ch[x][]) tagging(ch[x][],tag[x]);
    if(ch[x][]) tagging(ch[x][],tag[x]);
    tag[x]=;
}

void split(int now,int k,int &x,int &y)
{
    if(!now) x=y=;
    else
    {
        if(tag[now]) down(now);
        if(val[now]<=k)
        {
            x=now;
            split(ch[now][],k,ch[x][],y);
        }
        else
        {
            y=now;
            split(ch[now][],k,x,ch[y][]);
        }
        update(now);
    }
}

int merge(int x,int y)
{
    if(x && tag[x]) down(x);
    if(y && tag[y]) down(y);
    if(!x || !y)  return x+y;
    if(pri[x]<pri[y])
    {
        ch[x][]=merge(ch[x][],y);
        update(x);
        return x;
    }
    else
    {
        ch[y][]=merge(x,ch[y][]);
        update(y);
        return y;
    }
}

int main()
{
    int n;
    read(n);
    for(int i=;i<=n;++i) read(a[i]);
    for(int i=;i<=n;++i) read(t[i]);
    for(int i=;i<=n+;++i) id[i]=i;
    random_shuffle(id+,id+n+);
    int x,y,z;
    for(int i=;i<=n;++i)
    {
        //insert(a[i]);
        split(root,a[i],x,y);
        root=merge(merge(x,newnode(a[i])),y);
        split(root,t[i],x,y);
        cout<<sum[x]+1LL*t[i]*siz[y]<<' ';
        if(y) tagging(y,t[i]);
        root=y;
    }
    return ;
}