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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM