bzoj 3196: Tyvj 1730 二逼平衡树 树套树

→题目链接←

【想说的话】

不知道为什么这么晚了突然就想写了...

这是我第一颗树套树了，其实算半颗？有一半用pb_ds水的=.=

但是写出来还是很有成就感的...但是也同时也非常的虚平衡树...

【题解】

线段树套平衡树

用pbds搞得平衡树，这种东西需要编译器版本好高...bzoj上null_type还不行，必须换成旧版的null_mapped_type

（如果你的编译器不可编译，看我博客左边，有个链接，里面有教程）

【代码】

#include<bits/stdc++.h>#include<ext/pb_ds/tree_policy.hpp>#include<ext/pb_ds/assoc_container.hpp>using namespace std;using namespace __gnu_pbds;#define pr pair<int,int>#define N 50050#define inf 1000000000typedef tree<pr,null_mapped_type,less<pr>,rb_tree_tag,tree_order_statistics_node_update> rb_tree;typedef rb_tree::iterator ITER;inline int rd(){int x=0,y=1;char c=getchar();while(c<'0' || c>'9'){if(c=='-')y=-y;c=getchar();}while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();return x*y;}int n,m;rb_tree tr[N*4];int a[N],num[N],tot=0;void build(int x,int l,int r){if(l==r){tr[x].insert(pr(a[l],l));return;}int mid=(l+r)/2;build(x*2,l,mid);build(x*2+1,mid+1,r);tr[x]=tr[x*2];for(ITER i=tr[x*2+1].begin(); i!=tr[x*2+1].end(); i++)tr[x].insert(*i);}int query_rank(int x,int l,int r,int l1,int r1,int k,int kk){if(l>=l1 && r<=r1)return tr[x].order_of_key(pr(k,kk));int mid=(l+r)/2,ret=0;if(mid>=r1)ret=query_rank(x*2,l,mid,l1,r1,k,kk);else if(mid<l1)ret=query_rank(x*2+1,mid+1,r,l1,r1,k,kk);else ret=query_rank(x*2,l,mid,l1,mid,k,kk)+query_rank(x*2+1,mid+1,r,mid+1,r1,k,kk);return ret;}inline int query_kth(int l1,int r1,int k){int l=0,r=100000000;while(l<r){int mid=(l+r)/2;if(query_rank(1,1,n,l1,r1,mid,inf)>=k)r=mid;else l=mid+1;}return l;}inline void update(int x,int l,int r,int p,int k){tr[x].erase(pr(a[p],num[p]));tr[x].insert(pr(k,tot));if(l==r)return;int mid=(l+r)/2;if(p<=mid)update(x*2,l,mid,p,k);else update(x*2+1,mid+1,r,p,k);}int pre(int x,int l,int r,int l1,int r1,int k){if(l1<=l && r<=r1){ITER it=tr[x].lower_bound(pr(k,0));if(it==tr[x].begin())return -inf;it--;return (*it).first;}int mid=(l+r)/2,ret=-inf;if(r1<=mid)ret=pre(x*2,l,mid,l1,r1,k);else if(l1>mid)ret=pre(x*2+1,mid+1,r,l1,r1,k);else ret=max(pre(x*2,l,mid,l1,mid,k),pre(x*2+1,mid+1,r,mid+1,r1,k));return ret;}int nxt(int x,int l,int r,int l1,int r1,int k){if(l1<=l && r<=r1){ITER it=tr[x].upper_bound(pr(k,inf));if(it==tr[x].end())return inf;return (*it).first;}int mid=(l+r)/2,ret=inf;if(r1<=mid)ret=nxt(x*2,l,mid,l1,r1,k);else if(l1>mid)ret=nxt(x*2+1,mid+1,r,l1,r1,k);else ret=min(nxt(x*2,l,mid,l1,mid,k),nxt(x*2+1,mid+1,r,mid+1,r1,k));return ret;}int main(){n=rd(),m=rd();for(int i=1; i<=n; i++)a[i]=rd(),num[i]=i;tot=n;build(1,1,n);while(m--){int opt=rd();if(opt==1){int x=rd(),y=rd(),z=rd();printf("%d\n",query_rank(1,1,n,x,y,z,0)+1);}if(opt==2){int x=rd(),y=rd(),z=rd();printf("%d\n",query_kth(x,y,z));}if(opt==3){int x=rd(),y=rd();tot++;update(1,1,n,x,y);num[x]=tot;a[x]=y;}if(opt==4){int x=rd(),y=rd(),z=rd();printf("%d\n",pre(1,1,n,x,y,z));}if(opt==5){int x=rd(),y=rd(),z=rd();printf("%d\n",nxt(1,1,n,x,y,z));}}return 0;}