【bzoj2947】促销 平衡树

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=2947

【题解】

平衡树裸题，据说可以各种水过

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
struct data{ll l,r,v,size,fix,w;}tr[1000005];
ll T,rt,ans,len;
char buf[1<<15],*fs,*ft;
inline char getc() {return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
inline ll read()
{
    ll x=0,f=1;  char ch=getc();
    while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getc();}
    while(isdigit(ch))  {x=x*10+ch-'0';  ch=getc();}
    return x*f;
}
void update(ll k){tr[k].size=tr[tr[k].l].size+tr[tr[k].r].size+tr[k].w;}
void rturn(ll &k){ll t=tr[k].l;tr[k].l=tr[t].r;tr[t].r=k;tr[t].size=tr[k].size;update(k);k=t;}
void lturn(ll &k){ll t=tr[k].r;tr[k].r=tr[t].l;tr[t].l=k;tr[t].size=tr[k].size;update(k);k=t;}
void insert(ll &k,ll x)
{
    if(k==0){len++;k=len;tr[k].size=tr[k].w=1;tr[k].v=x;tr[k].fix=rand();return;}
    tr[k].size++;
    if(tr[k].v==x)tr[k].w++;
    else if(x>tr[k].v){insert(tr[k].r,x);if(tr[tr[k].r].fix<tr[k].fix)lturn(k);}
    else {insert(tr[k].l,x);if(tr[tr[k].l].fix<tr[k].fix)rturn(k);} 
}
void del(ll &k,ll x)
{
    if(k==0)return; 
    if(tr[k].v==x)
    {
        if(tr[k].w>1){tr[k].w--;tr[k].size--;return;}
        if(tr[k].l*tr[k].r==0)k=tr[k].l+tr[k].r;//有一个儿子为空
        else if(tr[tr[k].l].fix<tr[tr[k].r].fix) rturn(k),del(k,x);
        else lturn(k),del(k,x);
    }
    else if(x>tr[k].v) tr[k].size--,del(tr[k].r,x);
    else tr[k].size--,del(tr[k].l,x);
}
ll minn(ll p) {while(tr[p].l)  p=tr[p].l;return tr[p].v;}
ll maxx(ll p) {while(tr[p].r)  p=tr[p].r;return tr[p].v;}
int main()
{
    //freopen("cin.in","r",stdin);
    //freopen("cout.out","w",stdout);
    T=read();
    while(T--)
    {
        ll n=read();
        for(ll i=1;i<=n;i++)  {ll x=read();  insert(rt,x);}
        ll temp1=minn(rt),temp2=maxx(rt);
        ans+=temp2-temp1;
        del(rt,temp2);  del(rt,temp1);
    }
    printf("%lld\n",ans);
    return 0;
}