BZOJ3196 二逼平衡树 ZKW线段树套vector(滑稽)

我实在是不想再打一遍树状数组套替罪羊树了。。。

然后在普通平衡树瞎逛的时候找到了以前看过vector题解

于是我想:为啥不把平衡树换成vector呢???

然后我又去学了一下ZKW线段树

就用ZKW线段树套vector水过啦!!!

每个ZKW线段树的节点保存一个vector

操作1在分出的vector上查询比它小的数有多少个然后相加再加1

操作2二分再上操作1

操作3修改需要修改的节点的vector

操作4在分出vector上查询前驱取最大

操作5与操作4同理

luogu主站5772ms上卡过,开O2的话2296ms,bzoj上8196ms,COGS上直接过

// It is made by XZZ
#include<cstdio>
#include<algorithm>
#include<vector>
#define il inline
#define rg register
#define vd void
#define sta static
using std::vector;
typedef long long ll;
il int gi(){
	rg int x=0,f=1;rg char ch=getchar();
	while(ch<'0'||ch>'9')f=ch=='-'?-1:f,ch=getchar();
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return x*f;
}
const int maxn=50001;
int lb[maxn],W[maxn],n,N=65536,m;
typedef const int& fast;
vector<int>d[65537<<1];
vector<int>::iterator it;
il vd Insert(fast x,fast y){d[x].insert(upper_bound(d[x].begin(),d[x].end(),y),y);}
il vd Erase(fast x,fast y){d[x].erase(lower_bound(d[x].begin(),d[x].end(),y));}
il int Rank(fast x,fast y){return lower_bound(d[x].begin(),d[x].end(),y)-d[x].begin();}
il int Kth(fast x,fast y){return d[x][y-1];}
il int Prev(fast x,fast y){
	it=lower_bound(d[x].begin(),d[x].end(),y);
	if(it==d[x].begin())return -2147483647;
	return *--it;
}
il int Next(fast x,fast y){
	it=upper_bound(d[x].begin(),d[x].end(),y);
	if(it==d[x].end())return 2147483647;
	return *it;
}
il vd UPD1(rg int x,fast y){for(x+=N;x;x>>=1)Insert(x,y);}
il vd UPD2(rg int x,fast y){for(x+=N;x;x>>=1)Erase(x,y);}
il int max(fast a,fast b){return a>b?a:b;}
il int min(fast a,fast b){return a<b?a:b;}
il int RNK(rg int l,rg int r,fast k){
	sta int ret;ret=1;
	for(l+=N-1,r+=N+1;l^r^1;l>>=1,r>>=1){
		if(~l&1)ret+=Rank(l^1,k);
		if(r&1)ret+=Rank(r^1,k);
	}return ret;
}
il int PRE(rg int l,rg int r,fast k){
	sta int ret;ret=-2147483647;
	for(l+=N-1,r+=N+1;l^r^1;l>>=1,r>>=1){
		if(~l&1)ret=max(ret,Prev(l^1,k));
		if(r&1)ret=max(ret,Prev(r^1,k));
	}return ret;
}
il int NXT(rg int l,rg int r,fast k){
	sta int ret;ret=2147483647;
	for(l+=N-1,r+=N+1;l^r^1;l>>=1,r>>=1){
		if(~l&1)ret=min(ret,Next(l^1,k));
		if(r&1)ret=min(ret,Next(r^1,k));
	}return ret;
}
int main(){
	n=gi(),m=gi();
	for(N=1;N<n;N<<=1)
	for(rg int i=1;i<=n;++i)lb[i]=i&-i;
	for(rg int i=1;i<=n;++i){
		W[i]=gi();
		UPD1(i,W[i]);
	}
	int opt,l,r,k;
	while(m--){
		opt=gi();
		if(opt==3)l=gi(),k=gi();
		else l=gi(),r=gi(),k=gi();
		if(opt==1)printf("%d\n",RNK(l,r,k));
		else if(opt==2){
			sta int ll,rr,mid;
			ll=0,rr=100000000;
			while(ll<rr){
				mid=((ll+rr)>>1)+1;
				if(RNK(l,r,mid)>k)rr=mid-1;
				else ll=mid;
			}printf("%d\n",ll);
		}else if(opt==3)UPD2(l,W[l]),UPD1(l,W[l]=k);
		else if(opt==4)printf("%d\n",PRE(l,r,k));
		else if(opt==5)printf("%d\n",NXT(l,r,k));
		else while(1);
	}
	return 0;
}
posted @ 2018-01-06 08:20  菜狗xzz  阅读(1724)  评论(1编辑  收藏  举报