Codeforces 702F T-shirt 可持久化Treap练手

Description

给出$n$件$T$-$shirt$的重要程度$q_i$和花费$c_i$，有$k$个人最开始分别有$b_i$的金钱，每个人的选衣服的策略都是一样的：将所有$T$-$shirt$按照重要程度$q$从大到小排序，重要程度相同的按花费从小到大排，然后每个人从头开始取$T$-$shirt$，如果金钱数大于当前的$T$-$shirt$的花费，那么就买下这件衣服，问每个人最多能够买的$T$-$shirt$数量。

Data Constraint

$1$≤$n$,$k$≤$2·10^5$ $1$≤$c_i$,$q_i$≤$10^9$ $1$≤$b_j$≤$10^9$

最暴力的做法

将所有的$T$-$shirt$按照题目要求排序，对于每一个人，从前往后扫，能买就买，不能买就跳过，最后算出答案。

可持久化Treap

最近学了可持久化$Treap$，便找了道题来练练手。
上述做法是$O$($nk$)，显然会超时。
将所有的人按照$b_i$排序，建出一棵$Treap$。
然后按顺序枚举每一件$T$-$shirt$，对于当前的第$i$件$T$-$shirt$，将$Treap$ $split$成两棵$Treap$，记为$T_1$和$T_2$。$T_1$里的$b_i$全部小于$c_i$(即买不起这件$T$-$shirt$的人)，$T_2$里的$b_i$全部大于等于$c_i$，于是便给$T_2$打上一个标记，将$T_2$内的全部$b_i$减掉$c_i$，答案加$1$。
如果我们这是便简单地将两棵$Treap$合并为一棵，便很有可能破坏序列的有序性。
因而，在合并之前，我们还需调整$T_1$和$T_2$，不停调整直到使得$T_1$中的最大值小于等于$T_2$中的最小值，随后才能合并。
对于每次调整，找出$T_2$内的最小值（记为 $k$），与$T_1$内的最大值（记为 $p$）比较，若$k$<$p$则暴力用$O$($log$ $n$)的时间将$k$从$T_2$中$split$掉，再用$split$和$merge$操作把$k$加入$T_1$中。
这样做看上去好像会超时，但我们分析一下复杂度，便能发现这是$O$($n$ $log_2^2$ $n$)的。
证明:
设$k$在被减掉$c_i$前的金钱数为$v$（$v$=$k$+$c_i$），显然有$c_i$>$p$，减掉$c_i$后便有$p$>$v$-$c_i$，
于是便有$c_i$>$p$>$v$-$c_i$，整理得$c_i$>$v\over 2$
这也就意味着每一次$v$减完$c_i$后会被调整(从$T_2$调整到$T_1$)，当且仅当$v$减去了一个大于自己一半的$c_i$，因而对于每一个$b_i$，最多会被暴力调整$log$ $n$次，调整一次是$O$($log$ $n$)的，因而总的时间复杂度就是$O$($n$ $log_2^2$ n <script type="math/tex" id="MathJax-Element-19102">n</script>)。

Code

#include<iostream> 
#include<cstdio>
#include<cstring>
#pragma GCC optimize(3) 
#include<algorithm> 
#include<ctime>

#define fo(i,j,l) for(int i=j;i<=l;i++)
#define fd(i,j,l) for(int i=j;i>=l;i--) 
#define fi first
#define se second
#define random(x) rand()%x

using namespace std;
typedef long long ll;
typedef pair<int,int> P;
const ll N=42e4,maxn=6e9;
struct note{
    int l,r,size;
    ll rnd;
}s[N];
struct nnn{
    ll v,c;
}gift[N],pe[N];

ll down[N],mo[N],va[N],h[N],dy[N];
int n,m,j,k,l,i,o,root;
ll ans[N];
void getdown(int o)
{
    mo[s[o].l]-=down[o];
    down[s[o].l]+=down[o];
    mo[s[o].r]-=down[o];
    down[s[o].r]+=down[o];
    va[s[o].l]+=h[o];
    va[s[o].r]+=h[o];
    h[s[o].l]+=h[o];
    h[s[o].r]+=h[o];
    down[o]=h[o]=0;
}

int merge(int a,int b)
{
    if(!(a*b))return a^b;
    if(!down[a])getdown(a);
    if(!down[b])getdown(b);
    if(s[a].rnd<s[b].rnd){
        int ls=merge(s[a].r,b);
        s[a].r=ls; 
        s[a].size=s[s[a].l].size+s[ls].size+1; 
        return a;
    }else{
        int ls=merge(a,s[b].l);
        s[b].l=ls;
        s[b].size=s[ls].size+s[s[b].r].size+1;
        return b;
    }
}

P split(int o,int wz)
{
    if(!wz)return P(0,o);
    if(down[o]!=0)getdown(o);
    if(s[s[o].l].size>=wz){
        P ls=split(s[o].l,wz);
        s[o].l=ls.se;
        s[o].size=s[s[o].r].size+s[ls.se].size+1;
        return P(ls.fi,o);
    }else{
        P ls=split(s[o].r,wz-s[s[o].l].size-1);
        s[o].r=ls.fi;
        s[o].size=s[s[o].l].size+s[ls.fi].size+1;
        return P(o,ls.se);
    }
}

int search(int o,ll p)
{
    if(o==0)return 0;
    if(down[o]!=0)getdown(o);
    if(mo[o]>=p) return search(s[o].l,p);
    else return s[s[o].l].size+1+search(s[o].r,p);
}

ll read()
{
    ll o=0; char ch=' ';
    for(;ch<'0'||ch>'9';ch=getchar());
    for(;ch>='0'&&ch<='9';ch=getchar())o=o*10+ch-48;
    return o;
}

void pri(ll o)
{
    if(!o)return;
    pri(o/10); putchar('0'+o%10);
}

void write(ll o)
{
    if(o==0)putchar('0');
    pri(o);
}

bool kmp(nnn a,nnn b){
    return a.v!=b.v ? a.v>b.v : a.c<b.c;
}

bool kmp2(nnn a,nnn b){
    return a.v<b.v;
}

int main()
{
    cin>>n;
    fo(i,1,n)gift[i].c=read(),gift[i].v=read();
    sort(gift+1,gift+n+1,kmp);
    cin>>m;
    fo(i,1,m)pe[i].v=read(),pe[i].c=i;
    sort(pe+1,pe+m+1,kmp2);
    fo(i,1,m)mo[i]=pe[i].v;
    srand(time(0));
    fo(i,1,m){
        s[i].size=1;
        s[i].rnd=random(maxn);
        if(i==1)root=1;else root=merge(root,i);
    }
    ll zd=0;
    fo(i,1,n){
        int k=search(root,gift[i].c);
        if(k==m)continue;
        if(k==0){
            mo[root]-=gift[i].c;
            va[root]+=1;
            down[root]+=gift[i].c;
            h[root]+=1;
            continue;
        }
        P bl=split(root,k);
        int le=bl.fi,ri=bl.se;
        mo[ri]-=gift[i].c;
        va[ri]+=1;
        down[ri]+=gift[i].c;
        h[ri]+=1;
        if(k!=1){
            P rr=split(le,k-1);
            zd=mo[rr.se];
            le=merge(rr.fi,rr.se);
        }else zd=mo[le];
        while(s[ri].size>0){
            P oo=split(ri,1);
            int zb=oo.fi,yb=oo.se;
            if(mo[zb]>=zd){
                ri=merge(zb,yb);
                break;
            }
            int y=search(le,mo[zb]);
            P kk=split(le,y);
            if(y!=0)zb=merge(kk.fi,zb);
            if(kk.se!=0)zb=merge(zb,kk.se);
            le=zb; s[0].size=0; ri=oo.se;
        }
        root=merge(le,ri);
    }
    int y;P kk;
    fo(i,1,m){
        if(i!=m)kk=split(root,1),y=kk.fi;
        else y=root;
        ans[pe[y].c]=va[y];
        root=kk.se;
    }
    fo(i,1,m)write(ans[i]),putchar(' ');
}