[P3332][ZJOI2013]K大数查询(树套树)

感觉跟Dynamic Rankings差不多,但是需要区间修改。可以考虑分别维护两个树状数组,Σroot[1~i]记录第i个位置目前都有哪些数,Σroot1[1~i]记录第i个位置分别和前面i-1个位置的差的和。这样修改时就差分变成了单点修改,查询时用root中的值*位置的序号-root1中的值求出从第1个到所求位置的数的个数和,再加上和上面那道题一样的树状数组和主席树的操作就可以了。

 

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=100010;
struct node{
	int l,r;
	ll x;
}tree[N*400];
int n,m,num,cnta,cntb,cntc,cntd,a1[N],b1[N],c1[N],d1[N],root[N],root1[N];
inline int read(){
	int x=0,f=0;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return f?-x:x;
}
inline ll read1(){
	ll x=0;char ch=getchar();
	while(ch<'0'||ch>'9')ch=getchar();
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x;
}
inline int getbit(int x){
	return x&(-x);
}
void add1(int& p,int pa,int pb,int x,int f){
	if(!p){p=++num;tree[num].x=0;}
	if(pa==pb){tree[p].x+=1ll*f;return;}
	int mid=(pa+pb)>>1;
	x<=mid?add1(tree[p].l,pa,mid,x,f):add1(tree[p].r,mid+1,pb,x,f);
	tree[p].x=tree[tree[p].l].x+tree[tree[p].r].x;
	return;
}
int query1(int pa,int pb,int a,int b,ll k){
	if(pa==pb)return pa;
	ll lft=0;int mid=(pa+pb)>>1;
	for(int i=1;i<=cntb;++i)lft+=tree[tree[b1[i]].l].x*b;
	for(int i=1;i<=cntd;++i)lft-=tree[tree[d1[i]].l].x;
	for(int i=1;i<=cnta;++i)lft-=tree[tree[a1[i]].l].x*a;
	for(int i=1;i<=cntc;++i)lft+=tree[tree[c1[i]].l].x;
	if(lft>=k){
		for(int i=1;i<=cnta;++i)a1[i]=tree[a1[i]].l;
		for(int i=1;i<=cntb;++i)b1[i]=tree[b1[i]].l;
		for(int i=1;i<=cntc;++i)c1[i]=tree[c1[i]].l;
		for(int i=1;i<=cntd;++i)d1[i]=tree[d1[i]].l;
		return query1(pa,mid,a,b,k);
	}else{
		for(int i=1;i<=cnta;++i)a1[i]=tree[a1[i]].r;
		for(int i=1;i<=cntb;++i)b1[i]=tree[b1[i]].r;
		for(int i=1;i<=cntc;++i)c1[i]=tree[c1[i]].r;
		for(int i=1;i<=cntd;++i)d1[i]=tree[d1[i]].r;
		return query1(mid+1,pb,a,b,k-lft);
	}
}
inline void add(int x,int p,int f){
	while(x<=n)add1(root[x],1,n<<1|1,p,f),x+=getbit(x);
}
inline void adddelta(int x,int p,int f){
	while(x<=n)add1(root1[x],1,n<<1|1,p,f),x+=getbit(x);
}
inline int query(int a,int b,ll k){
	cnta=cntb=cntc=cntd=0;int aa=a,bb=b;
	while(a)a1[++cnta]=root[a],c1[++cntc]=root1[a],a-=getbit(a);
	while(b)b1[++cntb]=root[b],d1[++cntd]=root1[b],b-=getbit(b);
	return query1(1,n<<1|1,aa,bb,k);
}
int main(){
	n=read(),m=read(),num=0;
	for(int q,a,b,i=1;i<=m;++i){
		q=read(),a=read(),b=read();
		if(q==1){int c=read();add(a,n+1-c,1),add(b+1,n+1-c,-1),adddelta(a,n+1-c,a-1),adddelta(b+1,n+1-c,-b);}
		else{ll c=read1();printf("%d\n",n+1-query(a-1,b,c));}
	}
	return 0;
}

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值