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 1000000000

typedef 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;
}