【BZOJ3196】二逼平衡树(树状数组,线段树)

【BZOJ3196】二逼平衡树(树状数组,线段树)

题面

BZOJ题面

题解

如果不存在区间修改操作:
搞一个权值线段树
区间第K大--->直接在线段树上二分
某个数第几大--->查询一下区间的size和
某个数的前缀--->先查一下他是区间第几大,再求他-1大
某个数的后缀--->和上面那个有区别吗???

现在有了区间修改操作
多搞一个树状数组
套在一起就好啦
暴力开点开不下的
要动态开点

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define MAX 100000
inline int read()
{
	int x=0,t=1;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=-1,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return x*t;
}
struct Node
{
	int ls,rs;
	int v;
}t[MAX<<7];
int tot,S[MAX],rt[MAX],n,m,a[MAX];
int t1,t2,tmp1[MAX],tmp2[MAX];
int opt[MAX],ll[MAX],rr[MAX],kk[MAX];
int Tot;
inline int lowbit(int x){return x&(-x);}
void Modify(int &now,int l,int r,int pos,int w)
{
	if(!now)now=++Tot;
	t[now].v+=w;
	if(l==r)return;
	int mid=(l+r)>>1;
	if(pos<=mid)Modify(t[now].ls,l,mid,pos,w);
	else Modify(t[now].rs,mid+1,r,pos,w);
}
void PreModify(int pos,int w){for(int i=pos;i<=n;i+=lowbit(i))Modify(rt[i],1,tot,a[pos],w);}
int Query(int now,int l,int r,int pos)
{
	if(l==r)return t[now].v;
	int mid=(l+r)>>1;
	if(pos<=mid)return Query(t[now].ls,l,mid,pos);
	else return t[t[now].ls].v+Query(t[now].rs,mid+1,r,pos);
}
int Rank(int l,int r,int k)
{
	if(l>r)return 0;
	l-=1;int ret=0;
	for(int i=r;i;i-=lowbit(i))ret+=Query(rt[i],1,tot,k);
	for(int i=l;i;i-=lowbit(i))ret-=Query(rt[i],1,tot,k);
	return ret;
}
int Kth(int l,int r,int k)
{
	if(l==r)return l;
	int sum=0,mid=(l+r)>>1;
	for(int i=1;i<=t1;++i)sum+=t[t[tmp1[i]].ls].v;
	for(int i=1;i<=t2;++i)sum-=t[t[tmp2[i]].ls].v;
	if(sum>=k)
	{
		for(int i=1;i<=t1;++i)tmp1[i]=t[tmp1[i]].ls;
		for(int i=1;i<=t2;++i)tmp2[i]=t[tmp2[i]].ls;
		return Kth(l,mid,k);
	}
	else
	{
		for(int i=1;i<=t1;++i)tmp1[i]=t[tmp1[i]].rs;
		for(int i=1;i<=t2;++i)tmp2[i]=t[tmp2[i]].rs;
		return Kth(mid+1,r,k-sum);
	}
}
int PreKth(int l,int r,int k)
{
	l--;t1=t2=0;
	for(int i=r;i;i-=lowbit(i))tmp1[++t1]=rt[i];
	for(int i=l;i;i-=lowbit(i))tmp2[++t2]=rt[i];
	return S[Kth(1,tot,k)];
}
int main()
{
	n=read();m=read();
	for(int i=1;i<=n;++i)a[i]=S[++tot]=read();
	for(int i=1;i<=m;++i)
	{
		opt[i]=read();
		if(opt[i]!=3)
		{
			ll[i]=read();rr[i]=read();kk[i]=read();
			if(opt[i]!=2)S[++tot]=kk[i];
		}
		else
		{
			ll[i]=rr[i]=read();
			S[++tot]=kk[i]=read();
		}
	}
	sort(&S[1],&S[tot+1]);
	tot=unique(&S[1],&S[tot+1])-S-1;
	for(int i=1;i<=n;++i)a[i]=lower_bound(&S[1],&S[tot+1],a[i])-S,PreModify(i,1);
	for(int i=1;i<=m;++i)
	{
		if(opt[i]==1)
		{
			kk[i]=lower_bound(&S[1],&S[tot+1],kk[i])-S;
			printf("%d\n",Rank(ll[i],rr[i],kk[i]-1)+1);
		}
		else if(opt[i]==2)
		{
			printf("%d\n",PreKth(ll[i],rr[i],kk[i]));
		}
		else if(opt[i]==3)
		{
			PreModify(ll[i],-1);
			a[ll[i]]=lower_bound(&S[1],&S[tot+1],kk[i])-S;
			PreModify(ll[i],1);
		}
		else if(opt[i]==4)
		{
			kk[i]=lower_bound(&S[1],&S[tot+1],kk[i])-S;
			int gg=Rank(ll[i],rr[i],kk[i]-1);
			if(!gg)puts("-2147483647");
			else printf("%d\n",PreKth(ll[i],rr[i],gg));
		}
		else
		{
			kk[i]=lower_bound(&S[1],&S[tot+1],kk[i])-S;
			int gg=Rank(ll[i],rr[i],kk[i]);
			if(gg==rr[i]-ll[i]+1)puts("2147483647");
			else printf("%d\n",PreKth(ll[i],rr[i],gg+1));
		}
	}
	return 0;
}

posted @ 2018-01-08 16:40  小蒟蒻yyb  阅读(317)  评论(2编辑  收藏  举报