bzoj 3196 Tyvj 1730 二逼平衡树

Key To Problem

Code

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define N 50010
#define inf 1000000000
#define ls tr[k].l
#define rs tr[k].r
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
using namespace std;
struct node
{
int l,r,v,w,size,rnd;
};
node tr[N<<6];
int n,m,size,ans;
int a[N<<2];
int root[N<<2];

void updata(int &k)
{
tr[k].size=tr[ls].size+tr[rs].size+tr[k].w;
}

void lturn(int &k)
{
int t=rs;
tr[k].r=tr[t].l,tr[t].l=k;
tr[t].size=tr[k].size;
updata(k);
k=t;
}

void rturn(int &k)
{
int t=ls;
tr[k].l=tr[t].r,tr[t].r=k;
tr[t].size=tr[k].size;
updata(k);
k=t;
}

void insert(int &k,int x)
{
if(!k)
{
k=++size;
tr[k].v=x,tr[k].w=tr[k].size=1,tr[k].rnd=rand();
return ;
}
tr[k].size++;
if(x==tr[k].v)tr[k].w++;
else if(x>tr[k].v)
{
insert(rs,x);
if(tr[rs].rnd<tr[k].rnd)lturn(k);
}else if(x<tr[k].v)
{
insert(ls,x);
if(tr[ls].rnd<tr[k].rnd)rturn(k);
}
}

void build(int l,int r,int rt,int c,int x)
{
insert(root[rt],c);
if(l==r)return ;
int mid=(l+r)>>1;
if(mid>=x)build(lson,c,x);
else build(rson,c,x);
}

void ask_rank(int &k,int x)
{
if(!k)return ;
if(tr[k].v==x)
{
ans+=tr[ls].size;
return ;
}else if(x>tr[k].v)
ans+=tr[ls].size+tr[k].w,ask_rank(rs,x);
else
ask_rank(ls,x);
}

void query_rank(int l,int r,int rt,int L,int R,int c)
{
if(l==L&&r==R)
{
ask_rank(root[rt],c);
return ;
}
int mid=(l+r)>>1;
if(mid>=R)query_rank(lson,L,R,c);
else if(mid<L)query_rank(rson,L,R,c);
else{
query_rank(lson,L,mid,c);
query_rank(rson,mid+1,R,c);
}
}

int query_index(int L,int R,int c)
{
int l=0,r=inf,tmp;
while(l<=r)
{
int mid=(l+r)>>1;
ans=1,query_rank(1,n,1,L,R,mid);
if(ans<=c)
{
tmp=mid;
l=mid+1;
}else
r=mid-1;
}
return tmp;
}

void del(int &k,int x)
{
if(x==tr[k].v)
{
if(tr[k].w>1)tr[k].w--,tr[k].size--;
else if(ls*rs==0)k=ls+rs;
else if(tr[ls].rnd<tr[rs].rnd)
rturn(k),del(k,x);
else lturn(k),del(k,x);
}else if(x>tr[k].v)
del(rs,x),tr[k].size--;
else del(ls,x),tr[k].size--;
}

void change(int l,int r,int rt,int x,int c)
{
del(root[rt],a[x]);
insert(root[rt],c);
if(l==r)return ;
int mid=(l+r)>>1;
if(mid>=x)change(lson,x,c);
else change(rson,x,c);
}

void ask_before(int k,int x)
{
if(!k)return ;
if(x>tr[k].v)
ans=max(ans,tr[k].v),ask_before(rs,x);
else ask_before(ls,x);
}

void query_before(int l,int r,int rt,int L,int R,int c)
{
if(l==L&&r==R)
{
ask_before(root[rt],c);
return ;
}
int mid=(l+r)>>1;
if(mid>=R)query_before(lson,L,R,c);
else if(mid<L)query_before(rson,L,R,c);
else{
query_before(lson,L,mid,c);
query_before(rson,mid+1,R,c);
}
}

void ask_after(int k,int x)
{
if(!k)return ;
if(x<tr[k].v)
{
ans=min(ans,tr[k].v);
ask_after(ls,x);
}else ask_after(rs,x);
}

void query_after(int l,int r,int rt,int L,int R,int c)
{
if(l==L&&r==R)
{
ask_after(root[rt],c);
return ;
}
int mid=(l+r)>>1;
if(mid>=R)query_after(lson,L,R,c);
else if(mid<L)query_after(rson,L,R,c);
else{
query_after(lson,L,mid,c);
query_after(rson,mid+1,R,c);
}
}

int main()
{
//  freopen("lll.in","r",stdin);
//  freopen("ppp.out","w",stdout);
cin>>n>>m;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
build(1,n,1,a[i],i);
}
while(m--)
{
int opt,x,y,z;
scanf("%d",&opt);
switch(opt)
{
case 1: scanf("%d%d%d",&x,&y,&z);
ans=1,query_rank(1,n,1,x,y,z);
printf("%d\n",ans);
break;
case 2: scanf("%d%d%d",&x,&y,&z);
printf("%d\n",query_index(x,y,z));
break;
case 3: scanf("%d%d",&x,&z);
change(1,n,1,x,z),a[x]=z;
break;
case 4: scanf("%d%d%d",&x,&y,&z);
ans=0,query_before(1,n,1,x,y,z);
printf("%d\n",ans);
break;
case 5: scanf("%d%d%d",&x,&y,&z);
ans=inf,query_after(1,n,1,x,y,z);
printf("%d\n",ans);
break;
}
}
return 0;
}